2.pytorch中升维度的操作及需要注意的点

二、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是怎么一回事

1. unsqueeze_和unsqueeze实现一样的功能, 区别在于unsqueeze_是in_place操作。
   即unsqueeze不会对使用unsqueeze的tensor进行改变,想要获取unsqueeze后的值必须赋予个新值；unsqueeze_则会对自己改变
2. in_place会直接对原始的张量进行改变，不用对原始张量值再次进行赋值操作，很多函数都有这个操作
3. 【注意】：pytorch中很多的函数都是这样类似的 原理，两个函数名称相同，不同点就在于其中一个函数名后面有_，表示可以不用重新赋值，直接对改变张量的值。