mars.tensor.
swapaxes
Interchange two axes of a tensor.
a (array_like) – Input tensor.
axis1 (int) – First axis.
axis2 (int) – Second axis.
a_swapped – If a is a Tensor, then a view of a is returned; otherwise a new tensor is created.
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]]])