二、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中很多的函数都是这样类似的 原理,两个函数名称相同,不同点就在于其中一个函数名后面有_
,表示可以不用重新赋值,直接对改变张量的值。