mlx.core.argsort

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

返回对数组进行排序的索引。

参数:

• a (array) – 输入数组。

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

返回:

包含对输入进行排序的索引的 uint32 数组。

返回类型:

array