mars.tensor.transpose#

mars.tensor.transpose(a, axes=None)[source]#

Permute the dimensions of a tensor.

Parameters

  • a (array_like) – Input tensor.

  • axes (list of ints, optional) – By default, reverse the dimensions, otherwise permute the axes according to the values given.

Returns

p – a with its axes permuted. A view is returned whenever possible.

Return type

Tensor

See also

moveaxis, argsort

Notes

Use transpose(a, argsort(axes)) to invert the transposition of tensors when using the axes keyword argument.

Transposing a 1-D array returns an unchanged view of the original tensor.

Examples

>>> import mars.tensor as mt

>>> x = mt.arange(4).reshape((2,2))
>>> x.execute()
array([[0, 1],
       [2, 3]])

>>> mt.transpose(x).execute()
array([[0, 2],
       [1, 3]])

>>> x = mt.ones((1, 2, 3))
>>> mt.transpose(x, (1, 0, 2)).shape
(2, 1, 3)