mlx.core.swapaxes

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

Swap two axes of an array.

参数:

• a (array) -- Input array.

• axis1 (int) -- Specifies the first axis.

• axis2 (int) -- Specifies the second axis.

返回:

The array with swapped axes.

返回类型:

array