pytorch中的Permute转换

permute(dims)

将tensor的维度换位。

参数：参数是一系列的整数，代表原来张量的维度。比如三维就有0，1，2这些dimension。

举个例子

import torch
import numpy    as np

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


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



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

print(permuted)

 

参考资料

PyTorch中permute的用法_York1996的博客-CSDN博客_permute pytorch

Pytorch view() permute() contiguous() transpose()_我是天才很好的博客-CSDN博客