PyTorch中permute的用法
permute(dims)将tensor的维度换位。
参数：参数是一系列的整数，代表原来张量的维度。比如三维就有0，1，2这些dimension。
例：
import torch
import numpy as np
a=np.array([[,]])
unpermuted=torch.tensor(a)
print(unpermuted.size())#——>torch.Size()
permuted=unpermuted.permute(2,0,1)
print(permuted.size()) #——>torch.Size()
再比如图片img的size比如是（28，28，3）就可以利用img.permute(2,0,1)得到一个size为（3，28，28）的tensor。
利用这个函数permute（0，2，1）可以把Tensor([[,]]) 转换成
tensor([[,
,
]])
https://blog.csdn.net/york1996/article/details/81876886
页:
[1]