mars.tensor.swapaxes#

mars.tensor.swapaxes(a, axis1, axis2)[source]#

Interchange two axes of a tensor.

Parameters

  • a (array_like) – Input tensor.

  • axis1 (int) – First axis.

  • axis2 (int) – Second axis.

Returns

a_swapped – If a is a Tensor, then a view of a is returned; otherwise a new tensor is created.

Return type

Tensor

Examples

>>> import mars.tensor as mt

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

>>> x = mt.array([[[0,1],[2,3]],[[4,5],[6,7]]])
>>> x.execute()
array([[[0, 1],
        [2, 3]],
       [[4, 5],
        [6, 7]]])

>>> mt.swapaxes(x,0,2).execute()
array([[[0, 4],
        [2, 6]],
       [[1, 5],
        [3, 7]]])