-
transpose와 permute카테고리 없음 2023. 3. 22. 16:27
permute()와 transpose()는 유사한 방식으로 작동한다.
transpose()는 딱 두 개의 차원을 맞교환할 수 있다.
그러나 permute()는 모든 차원들을 맞교환할 수 있다.
x = torch.rand(16, 32, 3) y = x.tranpose(0, 2) # [3, 32, 16] z = x.permute(2, 1, 0) # [3, 32, 16]