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]

>>阅读原文<<