torch.transpose

• 바꿀 배열과 바꿀 차원 두 개를 받고, 그 둘의 차원을 바꿈

예시

x = torch.rand(16, 32, 3)
y = x.tranpose(0, 2)  # [3, 32, 16]

0과 2의 차원을 바꿔서 그 둘의 차원이 바뀜.