參考：
permute函數(shù)
python pytorch permute函數(shù)
[Python] 維度交換函數(shù)：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)

總結(jié)

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

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。