python pytorch permute函数

permute(多维数组,[维数的组合])

比如:

a=rand(2,3,4);  %这是一个三维数组，各维的长度分别为：2，3，4

%现在交换第一维和第二维：

permute(A,[2,1,3])  %变成3*2*4的矩阵

import torch
import numpy    as np

a=np.array([[[1,2,3],[4,5,6]]])


unpermuted=torch.tensor(a)
print(unpermuted.size())  #  ——>  torch.Size([1, 2, 3])

tensor([[[1., 4.],
        [2., 5.],
        [3., 6.]]])


permuted=unpermuted.permute(2,0,1)
print(permuted.size())     #  ——>  torch.Size([3, 1, 2])


tensor([[[1., 2.],
         [3., 4.],
         [5., 6.]]])