torch.squeeze
作用：去除size為1的維度。當維度大于等于2時，squeeze()無作用。

>>> x = torch.zeros(2, 1, 2, 1, 2) >>> x.size() torch.Size([2, 1, 2, 1, 2]) # 可以看出size=1的維度有第二維（對應下標1）和第四維（對應下標3）>>> y = torch.squeeze(x) # 去除size為1的維度，即第二維和第四維 >>> y.size() torch.Size([2, 2, 2])>>> y = torch.squeeze(x, 0) # 第一維size為2，squeeze無作用 >>> y.size() torch.Size([2, 1, 2, 1, 2])>>> y = torch.squeeze(x, 1) # 去除第二維 >>> y.size() torch.Size([2, 2, 1, 2])>>> y = torch.squeeze(x, 2) # 第三維size為2，squeeze無作用 >>> y.size() torch.Size([2, 1, 2, 1, 2])>>> y = x.squeeze(-1) # 如果x的最后一維size為1，則去掉 >>> y.size() torch.Size([2, 1, 2, 1, 2])

torch.unsqueeze
作用：與squeeze()作用相反，用于添加維度。unsqueeze必須指明維度

>>> x = torch.zeros(2, 1, 3, 4, 2) >>> x.size() torch.Size([2, 1, 3, 4, 2])>>>y = torch.unsqueeze(x,0) # 在x的第0維前添加一個維度 >>>y.size() torch.Size([1, 2, 1, 3, 4, 2])>>>y = torch.unsqueeze(x,1) # 在x的第1維前添加一個維度 >>>y.size() torch.Size([2, 1, 1, 3, 4, 2])

總結(jié)

以上是生活随笔為你收集整理的torch.squeeze和torch.unsqueeze的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。