[PyTorch] torch.transpose()와 numpy().transpose() 차이

파이토치의 transpose는 N $\times$ M 만 transpose 할 수 있다는 단점이 있다.

그러면 N $\times$ M $\times$ K는 어떻게 나타낼 수 있을까?

=> numpy로 바꾼 후 transpose() 하면 된다.

 

import torch
import numpy as np
x = torch.rand(3,4,2)
print(x.numpy().transpose((1,2,0)))