标签:中升 升维 image unsqueeze 张量 pytorch place 维度 13
二、unsqueeze、unsqueeze_和None的使用及注意in_place操作
1.unsqueeze
unsqueeze
没有in_place
,就不可以直接对原始张量进行赋值,需要重新对张量进行赋值。
image = torch.randn(3, 13, 13)
# 升维,变成NCHW
image = image.unsqueeze(dim=0)
2.unsqueeze_
unsqueeze_
有in_place
,可以直接对原始张量进行赋值。
image = torch.randn(3, 13, 13)
# 升维,变成NCHW
image.unsqueeze_(dim=0) # 这里会直接改变张量的值
3.None的使用
升维可以使用None
和...
来进行升维操作,同时需要重新对张量赋值。
image =torch.randn(3, 13, 13)
# 升维,变成NCHW
image = image[None, ...]
4. 注意,in_place是怎么一回事
unsqueeze_
和unsqueeze
实现一样的功能, 区别在于unsqueeze_
是in_place
操作。
即unsqueeze
不会对使用unsqueeze
的tensor进行改变,想要获取unsqueeze后的值必须赋予个新值;unsqueeze_
则会对自己改变in_place
会直接对原始的张量进行改变,不用对原始张量值再次进行赋值操作,很多函数都有这个操作【注意】
:pytorch中很多的函数都是这样类似的 原理,两个函数名称相同,不同点就在于其中一个函数名后面有_
,表示可以不用重新赋值,直接对改变张量的值。
标签:中升,升维,image,unsqueeze,张量,pytorch,place,维度,13 来源: https://blog.csdn.net/weixin_45745378/article/details/108189419
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。