Pytorch的tensor操作

• permute()函数

permute()函数

是对tensor的维度位置进行变换
具体理解看代码

a = torch.rand(1, 2, 3)
print(a.shape)
a = a.permute(2, 1, 0)
print(a.shape)

torch.Size([1, 2, 3])
torch.Size([3, 2, 1])