參考：
permute函數
python pytorch permute函數
[Python] 維度交換函數：transpose(m,n,r)和permute(m,n,r)

import torch from torch import nn import numpy as np b=np.array([[[ 0, 1, 2,3],[ 4,5,6,7],[ 8,9, 10,11]],[[12, 13, 14, 15],[16, 17,18, 19],[20, 21,22, 23]]]) print(b) [[[ 0 1 2 3][ 4 5 6 7][ 8 9 10 11]][[12 13 14 15][16 17 18 19][20 21 22 23]]] unpermuted=torch.tensor(b) permuted = unpermuted.permute(2,1,0) print(permuted) tensor([[[ 0, 12],[ 4, 16],[ 8, 20]],[[ 1, 13],[ 5, 17],[ 9, 21]],[[ 2, 14],[ 6, 18],[10, 22]],[[ 3, 15],[ 7, 19],[11, 23]]], dtype=torch.int32)

總結

以上是生活随笔為你收集整理的torch.tensor().permute(2,1,0)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。