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

cupy.ndarray

Note

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.