函數功能：将tensor的维度换位
参数：参数是一系列的整数，代表原来张量的维度。比如三维就有0，1，2这些dimension。
例：
复制代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16import 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内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复