torch.squeeze：將tensor中大小為1的維度刪除
torch.squeeze(tensor,dim）
代碼示例：

import torch a=torch.ones(2,2,2,1,1) b=torch.squeeze(a) c=torch.squeeze(a,0) d=torch.squeeze(a,3) print("a的shape：",a.shape) print("b的shape：",b.shape) print("c的shape：",c.shape) print("d的shape：",d.shape)

輸出：

a的shape： torch.Size([2, 2, 2, 1, 1]) b的shape： torch.Size([2, 2, 2]) c的shape： torch.Size([2, 2, 2, 1, 1]) d的shape： torch.Size([2, 2, 2, 1])

注：對特定維度進行squeeze，如果該維度大小不為1，則保持原來的shape不變，如代碼中的c的shape。

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。