cupy.argpartition(a, kth, axis=-1)[source]

Returns the indices that would partially sort an array.

  • 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.


Array of the same type and shape as a.

Return type



For its implementation reason, cupy.argpartition fully sorts the given array as cupy.argsort does. It also does not support kind and order parameters that numpy.argpartition supports.