例如：

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

以上這篇pytorch 轉換矩陣的維數位置方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持武林站長站。