728x90 transpose1 [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))) 2023. 1. 1. 이전 1 다음 728x90
