megengine.functional.swapaxes

swapaxes(inp, axis1, axis2)

交换一个 tensor 的两个坐标轴。

参数:

• inp (Tensor) – 要进行坐标轴交换的输入 tensor。

• axis1 (int) – 第一个坐标轴。

• axis2 (int) – 第二个坐标轴。

返回类型:

Tensor

返回:

‘inp’ 交换过指定坐标轴之后的 tensor。

实际案例

>>> x = Tensor(np.array([[[0,1],[2,3]],[[4,5],[6,7]]], dtype=np.int32))
>>> F.swapaxes(x, 0, 2)
Tensor([[[0 4]
 [2 6]]
[[1 5]
 [3 7]]], dtype=int32, device=xpux:0)