p
將tensor的維度換位。
引數：引數是一系列的整數，代表原來張量的維度。比如三維就有0，1，2這些dimension。
例：import torch
import numpy as np
a=np.array([[[1,2,3],[4,5,6]]])
unpermuted=torch.tensor(a)
print(unpermuted.size()) # ——> torch.size([1, 2, 3])www.cppcns.com
permuted=unpermuted.permute(2,0,1)
print(permuted.size()) # ——> torch.size([3, 1, 2])
再比如img的size比如是（28，28，cbxkdmjn3）就可以利用img.permute(2,0,1)得到乙個size為（3，28，28）的tensor。
利用這個函式permute（1，3，2）可以把tensor([[[1,2,3],[4,5,6]]]) 轉換成
tensor([[[1., 4.],
[2., 5.],
[3., 6.]]])
如果使用view（1,3,2），可以得到
tensor([[[1., 2.],
[3., 4.],
[5., 6.]]])
本文標題: pytorch中permute的用法詳解
本文位址: /jiaoben/python/296113.html
PyTorch中permute的用法
permute dims 將tensor的維度換位。引數 引數是一系列的整數，代表原來張量的維度。比如三維就有0，1，2這些dimension。例 import torch import numpy as np a np.array 1,2,3 4,5,6 unpermuted torch.tens...
PyTorch中permute的用法詳解
函式功能 將tensor的維度換位 引數 引數是一系列的整數，代表原來張量的維度。比如三維就有0，1，2這些dimension。例 import torch import numpy as np a np.array 1,2,3 4,5,6 unpermuted torch.tensor a pri...
permute（） 對維度進行換位
將tensor的維度換位。引數 引數是一系列的整數，代表原來張量的維度。比如三維就有0，1，2這些dimension。例 import torch import numpy as np a np.array 1,2,3 4,5,6 unpermuted torch.tensor a print un...