mlx.core.argsort

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

回傳可排序陣列的索引。

參數:

• a (array) -- 輸入陣列。

• axis (int or None, optional) -- 可選的排序軸。若為 None，則對展平後的陣列排序。若未指定，預設為 -1（對最後一軸排序）。

回傳:

包含可排序輸入的索引之 uint32 陣列。

回傳型別:

array