mindspore.numpy.swapaxes

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

交换Tensor的两个轴。

参数：

  • x (Tensor) - 要调换的Tensor。

  • axis1 (int) - 第一个轴。

  • axis2 (int) - 第二个轴。

返回：

调换后的Tensor，其数据类型与原始Tensor x 相同。

异常：

  • TypeError - 如果 axis1 或 axis2 不是整数，或 x 不是Tensor。

  • ValueError - 如果 axis1 或 axis2 不在 \([-ndim, ndim-1]\) 的范围内。

支持平台：

Ascend GPU CPU

样例：

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