mindspore.Tensor.swapaxes

mindspore.Tensor.swapaxes(axis1, axis2)[源代码]

交换Tensor的两个维度。

参数:
  • axis1 (int) - 第一个维度。

  • axis2 (int) - 第二个维度。

返回:

转化后的Tensor,与输入具有相同的数据类型。

异常:
  • TypeError - axis1axis2 不是整数。

  • ValueError - axis1axis2 不在 [-ndim, ndim-1] 范围内。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.ones((2,3,4), dtype=np.float32))
>>> output = x.swapaxes(0, 2)
>>> print(output.shape)
(4,3,2)