Pytorch permute()的简单用法

作用：调换Tensor中各维度的顺序

输入：数组，用来表示调换后的各维度顺序。一开始没修改的顺序其实就是[0，1，2…]

例子：有一个6×7×8×9的Tensor，我们想将其第2维(8)和第三维(9)给调换顺序

import torch

x = torch.empty([6, 7, 8, 9])
print(x.size())
x = x.permute([0, 1, 3, 2])
print(x.size())

输出：

torch.Size([6, 7, 8, 9])
torch.Size([6, 7, 9, 8])