[tips] transpose and view function in pytorch

[jo-member]

a = torch.tensor([[[[1,2],[3,4]],[[5,6],[7,8]]]])
# torch.Size([1, 2, 2, 2])

b = a.view(1,2,4).transpose(1,2)
'''
tensor([[[1, 5],
         [2, 6],
         [3, 7],
         [4, 8]]])
'''
c = b.view(1,2,2,2)
'''
tensor([[[[1, 5],
          [2, 6]],
         [[3, 7],
          [4, 8]]]])
'''

c = b.transpose(1,2).view(1,2,2,2)
'''
tensor([[[[1, 2],
          [3, 4]],
         [[5, 6],
          [7, 8]]]])
'''

이렇게 tranpose를 해준뒤 view function을 사용해야 원래의 space대로 복구가 됩니다.

[remaindere]

1 따봉