python pytorch permute関数

permute(多次元配列,[次元の組合せ])
例:
a=rand(2,3,4);%これは3 D配列で、各次元の長さはそれぞれ:2,3,4です.
%では、第1次元と第2次元を交換します.
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.]]])