标签：unsqueeze tensor 张量 print PyTorch 维度 squeeze

pytorch中squeeze()和unsqueeze()作用

squeeze()

squeeze() 用于在张量的指定维度插入新的维度 (为1) 得到维度提升的张量。

unsqueeze()

unsqueeze() 用于移除指定或者所有维度大小为1的维度，从而得到维度减小的新张量。

演示代码：

import torch

a=torch.arange(0,6)
b=a.view(2,3) # 维度：2*3
print("b",b)

c=b.unsqueeze(0) # 维度：1*2*3
print("c",c)

d=c.unsqueeze(2) # 维度：1*2*1*3
print("d",d)

e=d.squeeze(0) # 维度：2*1*3
print("e",e)

f=e.squeeze(0) # 维度：2*1*3
print("f",f)

结果显示：

b tensor([[0, 1, 2],
        [3, 4, 5]])
c tensor([[[0, 1, 2],
         [3, 4, 5]]])
d tensor([[[[0, 1, 2]],

         [[3, 4, 5]]]])
e tensor([[[0, 1, 2]],

        [[3, 4, 5]]])
f tensor([[[0, 1, 2]],

        [[3, 4, 5]]])

标签：unsqueeze,tensor,张量,print,PyTorch,维度,squeeze
来源： https://blog.csdn.net/weixin_48622537/article/details/115073784

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。