mlx.core.moveaxis

moveaxis(a: array, /, source: int, destination: int, *, stream: None | Stream | Device = None) → array

将轴移动到新位置。

参数:

• a (array) – 输入数组。

• source (int) – 指定源轴。

• destination (int) – 指定目标轴。

返回:

轴已移动的数组。

返回类型:

array