维度变换相关函数主要有 torch.reshape(或者调用张量的view方法), torch.squeeze, torch.unsqueeze, torch.transpose
torch.reshape 可以改变张量的形状。
torch.squeeze 可以减少维度。
torch.unsqueeze 可以增加维度。
torch.transpose 可以交换维度。
import torch
torch.manual_seed(0)
minval,maxval = 0, 255
a = (minval + (maxval - minval)*torch.rand([1,3,3,2])).int()
print(a.shape)
print(a