mlx.core.partition

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

Returns a partitioned copy of the array such that the smaller kth elements are first.

The ordering of the elements in partitions is undefined.

參數:

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

• kth (int) -- Element at the kth index will be in its sorted position in the output. All elements before the kth index will be less or equal to the kth element and all elements after will be greater or equal to the kth element in the output.

• axis (int or None, optional) -- Optional axis to partition over. If None, this partitions over the flattened array. If unspecified, it defaults to -1.

回傳值:

The partitioned array.

回傳值型別:

array