permute

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

Return a new tensor with the specific order of dimensions.

Parameters:
Returns:

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