mindspore.Tensor.transpose

mindspore.Tensor.transpose(*axes)[源代码]

返回被转置后的Tensor。

  • 对于一维Tensor，这没有影响，因为转置后的向量是相同的。

  • 对于二维Tensor，是标准的矩阵转置。

  • 对于n维Tensor，如果提供了维度，则它们的顺序代表维度的置换方式。

如果未提供轴，且Tensor.shape等于(i[0], i[1],…i[n-2], i[n-1])，则Tensor.transpose().shape等于(i[n-1], i[n-2], … i[1], i[0])。

参数：

  • axes (Union[None, tuple(int), list(int), int], 可选) - 如果 axes 为None或未设置，则该方法将反转维度。如果 axes 为tuple(int)或list(int)，则Tensor.transpose()把Tensor转置为新的维度。如果 axes 为整数，则此表单仅作为元组/列表表单的备选。

返回：

Tensor，具有与输入Tensor相同的维度，其中维度被准确的排列。

异常：

  • TypeError - 输入参数类型有误。

  • ValueError - axes 的数量不等于Tensor.ndim。

支持平台：

Ascend GPU CPU

样例：

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.ones((1,2,3), dtype=np.float32))
>>> x = x.transpose()
>>> print(x.shape)
(3, 2, 1)