cupy.argpartition¶
-
cupy.
argpartition
(a, kth, axis=-1)[source]¶ Returns the indices that would partially sort an array.
Parameters: - a (cupy.ndarray) – Array to be sorted.
- kth (int or sequence of ints) – Element index to partition by. If supplied with a sequence of k-th it will partition all elements indexed by k-th of them into their sorted position at once.
- axis (int or None) – Axis along which to sort. Default is -1, which means sort along the last axis. If None is supplied, the array is flattened before sorting.
Returns: Array of the same type and shape as
a
.Return type: Note
For its implementation reason, cupy.argpartition fully sorts the given array as cupy.argsort does. It also does not support
kind
andorder
parameters thatnumpy.argpartition
supports.See also