pytorch 中 permute()函数的用法

permute()函数用于维度变换。在图像处理中经常出现，比如说读取的图片是 numpy格式，是(h,w,c),但是torch处理的数据是(c,h,w)。所以这个函数经常使用

import torch
##符合均值为0，方差为1的正态分布
x=torch.randn(2,3,5)
print(x.size())
##维度变换
x=torch.permute(x,(2,0,1))
print(x.size())