mlx.core.argpartition

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

回傳可分割陣列的索引。

分割內元素的順序由索引給定，但其順序未定義。

參數:

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

• kth (int) -- 輸出中 kth 位置的元素索引會給出排序位置。kth 之前的所有索引對應的元素皆小於或等於 kth 位置的元素，而 kth 之後的所有索引對應的元素皆大於或等於 kth 位置的元素。

• axis (int or None, optional) -- 可選的分割軸。若為 None，則對展平後的陣列分割。若未指定，預設為 -1。

回傳:

包含可分割輸入的索引之 uint32 陣列。

回傳型別:

array