squeeze()：
squeeze(arg)表示第arg維的維度值為1，則去掉該維度。否則tensor不變。（即若tensor.shape()[arg] = 1，則去掉該維度）
unsqueeze()：
unsqueeze(arg)與squeeze(arg)作用相反，表示在第arg維增加一個維度值為1的維度。
常用操作：
參考：https://blog.csdn.net/lanse_zhicheng/article/details/79148678
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。
猜您在找
Pytorch中的squeeze()和unsqueeze()函數
[ Pytorch ] torch.squeeze() 和torch.unsqueeze()的用法
pytorch 中改變tensor維度(transpose)、拼接(cat)、壓縮(squeeze)詳解
pytorch中的size()、 squeeze()函數
torch.squeeze()和torch.unsqueeze()
pytorch 中tensor的加減和mul、matmul、bmm
[Pytorch]Pytorch中tensor常用語法
pytorch 中Tensor.uniform_代替numpy.random.uniform
pytorch中的標准化的tensor轉圖像
Pytorch中的torch.cat()函數 tensor拼接