[파이토치] permute 함수란?

torch.permute 함수는 텐서의 차원을 재배열하는 함수입니다. 아래와 같은형식으로 사용합니다. 

torch.permute(텐서,재배열차원)

예를 들어봅시다. 아래와 같은 텐서를 하나 정의하였습니다. 

x = torch.tensor([[1, 2], [3, 4], [5, 6]])
>>> x
tensor([[1, 2],
        [3, 4],
        [5, 6]])

permute 함수를 적용하여 0차원을 1차원으로, 1차원을 0차원으로 재배열합니다. 

>>> torch.permute(x,(1,0))
tensor([[1, 3, 5],
        [2, 4, 6]])