例如：

preds = to_numpy(preds)#preds是[2985x16x2]preds = preds.transpose(2, 1, 0)#preds[2x16x2985]

以上這篇pytorch 轉(zhuǎn)換矩陣的維數(shù)位置方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持VEVB武林網(wǎng)。