你即將離開本站

並前往https://www.daimajiaoliu.com/daima/479bb7bfc100408

Pytorch-tensor的维度变化 - 代码交流

In[19]: a.unsqueeze(-1).shape # 在-1之后插入一个维度Out[19]: torch.Size([4, 1, 28, 28, 1]) # 理解上可能 ... squeeze(index) 当index对应的dim为1，就产生作用。

確定！ 回上一頁

查詢 「Torch expand dim」的人也找了：

  1. Torch unsqueeze
    2.
  2. Torch expand
    3.
  3. Torch expand dim
    4.
  4. Torch squeeze
    5.
  5. Torchcat
    6.
  6. Torch size
    7.
  7. Pytorch dimension
    8.
  8. Torch reshape
    9.