permute

dragon.vm.torch.permute(
  input,
  dims,
  out=None
)[source]

Return a tensor with the new order of dimensions.

Parameters:
Returns:

dragon.vm.torch.Tensor – The output tensor.