PyTorch 1.8 中文 · torch. torch.squeeze. torch.squeeze(input, dim=None, *, out=None) → Tensor. 返回一个张量,其中大小为 1 的 input 的所有维都已删除。
確定! 回上一頁