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)