mlx.core.sort

sort(a: array, /, axis: None | int = -1, *, stream: None | Stream | Device = None) → array

返回数组的排序副本。

参数：

• a (array) – 输入数组。

• axis (int 或 None, 可选) – 可选的排序轴。如果为 None，则对扁平化数组进行排序。如果未指定，则默认为 -1（在最后一个轴上排序）。

返回值：

排序后的数组。

返回类型：

array