mindspore.mint.transpose

mindspore.mint.transpose(input, dim0, dim1)

交换Tensor的两个维度。

警告

这是一个实验性API，后续可能修改或删除。

参数：

• input (Tensor) - 输入Tensor。

• dim0 (int) - 第一个维度。

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

返回：

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

异常：

• TypeError - 如果 input 不是Tensor。

• TypeError - 如果 dim0 或者 dim1 不是整数。

• ValueError - 如果 dim0 或者 dim1 不在 \([-ndim, ndim -1]\) 范围内。

支持平台：

Ascend

样例：

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