transpose

dragon.transpose(
  inputs,
  perm=None,
  copy=True,
  **kwargs
)[source]

Permute the dimensions of input.

Examples:

# Provide the permutation for all axes
x = dragon.ones(shape=(2, 3, 4))
print(dragon.transpose(x, (0, 2, 1)).shape)  # (2, 4, 3)

# Or dimensions will be simply inverse
print(dragon.transpose(x).shape)  # (4, 3, 2)
Parameters:
  • inputs (dragon.Tensor) The input tensor.
  • perm (Union[Sequence[int], dragon.Tensor]], optional) The output permutation.
  • copy (bool, optional, default=True) Return a new tensor or call in-place.
Returns:

dragon.Tensor The output tensor.