torch.permute
torch.permute(input, dims) -> Tensor
Returns a view of the original tensor "input" with its dimensions
   permuted.
Parameters:
      * input (Tensor) -- the input tensor.
  * **dims** (*tuple of python:int*) -- The desired ordering of
    dimensions

-[ Example ]-



x = torch.randn(2, 3, 5)
x.size()
   torch.Size([2, 3, 5])
torch.permute(x, (2, 0, 1)).size()
   torch.Size([5, 2, 3])


