mars.tensor.swapaxes#
- mars.tensor.swapaxes(a, axis1, axis2)[源代码]#
Interchange two axes of a tensor.
- 参数
- 返回
a_swapped – If a is a Tensor, then a view of a is returned; otherwise a new tensor is created.
- 返回类型
Tensor
实际案例
>>> 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]]])