mlx.core.swapaxes

swapaxes(a: array, /, axis1: int, axis2: int, *, stream: None | Stream | Device = None) → array

交换数组的两个轴。

参数：

• a (array) – 输入数组。

• axis1 (int) – 指定第一个轴。

• axis2 (int) – 指定第二个轴。

返回：

交换了轴的数组。

返回类型：

array