mars.tensor.transpose#

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

改变张量的维度排列。

Parameters

  • a (array_like) – 输入张量。

  • axes (列表 of 整数, 可选) – 默认情况下，反转维度，否则根据给定的值重新排列轴。

Returns

p – a 的轴被排列。尽可能返回一个视图。

Return type

张量

另请参阅

moveaxis, argsort

备注

使用 transpose(a, argsort(axes)) 来反转张量的转置，当使用 axes 关键字参数时。

转置一维数组返回原始张量的未改变视图。

示例

>>> 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)