概述
函數功能：将tensor的维度换位
参数：参数是一系列的整数，代表原来张量的维度。比如三维就有0，1，2这些dimension。
例：
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, 2, 3],
[4, 5, 6]])
permuted=unpermuted.permute(2,0,1)
print(permuted.size()) # ——> torch.Size([3, 1, 2])
#tensor([[[1, 4]],
[[2, 5]],
[[3, 6]]])
再比如图片img的size比如是（28，28，3）就可以利用img.permute(2,0,1)得到一个size为（3，28，28）的tensor。
只是操作tensor並不能操作numpy
最后
以上就是无聊流沙为你收集整理的PyTorch中permute的用法详解的全部内容，希望文章能够帮你解决PyTorch中permute的用法详解所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复