Pytorch Tensor維度交換

使用transpose，只能交換某兩個維度

大於2維的tensor的話，也可以想象成二維矩陣的轉置操作，可以觀察下面的例子得出，之後會畫個圖說明。

>>> t = torch.arange(8).reshape(2,2,2)
>>> t
tensor([[[0, 1],
         [2, 3]],

        [[4, 5],
         [6, 7]]])

>> t.transpose(0,2)
tensor([[[0, 4],
         [2, 6]],

        [[1, 5],
         [3, 7]]])

>>> t.transpose(0,1)
tensor([[[0, 1],
         [4, 5]],

        [[2, 3],
         [6, 7]]])