python pytorch permute函数

permute(多维数组,[维数的组合])python 好比:数组 a=rand(2,3,4);  %这是一个三维数组，各维的长度分别为：2，3，4code %如今交换第一维和第二维：class permute(A,[2,1,3])  %变成3*2*4的矩阵import import torch import numpy    as np a=np.array([[[1,2,3],[4,5,6]

>>阅读原文<<