PyTorch中permute的用法详解

permute(dims)

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])
permuted=unpermuted.permute(2,0,1)
print(permuted.size()) # ——> torch.Size([3, 1, 2])

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

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