permute()和transpose()都是对轴进行操作\
import torch
from PIL import Image
import numpy as np
img=Image.open(r'C:\Users\Administrator\Pictures\QQ浏览器截图\QQ浏览器截图20190301211443.png')
a=np.array(img)
b=torch.Tensor(a)
print(a.shape)
print(b.shape)
a=np.transpose(a,(2,0,1))
b=b.permute(2,0,1)
print(a.shape)
print(b.shape)