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)
print(unpermuted.size()) # ——> torch.size([1, 2, 3])
#tensor([[1, 2, 3],
[4, 5, 6]])
permuted=unpermuted.permute(2,0,1)
print(permuted.size()) # ——> torch.size([3, 1, 2])
#tensor([[[1, 4]],
[[2, 5]],
[[3, 6]]])

只是操作tensor並不能操作numpy

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的用法詳解

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 ...

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...