pytorch中unsqueeze_（增加维度）的用法

unsqueeze_用在pytorch中增加维度
unsqueeze_（0）：在0轴上增加维度
unsqueeze_（1）：在1轴上增加维度
。
。
。

import torch
from PIL import Image
import numpy as np
from torchvision import transforms
img=Image.open(r'C:\Users\Administrator\Pictures\QQ浏览器截图\QQ浏览器截图20190301211443.png')
a=np.array(img)
b=torch.Tensor(a)
print(b.shape)
#方法1：unsqueeze_
b.unsqueeze_(0)
print(b.shape)

import torch
from PIL import Image
import numpy as np
from torchvision import transforms
img=Image.open(r'C:\Users\Administrator\Pictures\QQ浏览器截图\QQ浏览器截图20190301211443.png')
a=np.array(img)
print(a.shape)
#方法二：array变成tensor的时候，加一个[]，可以直接加一个维度
b=torch.Tensor([a])
print(b.shape)