cunumeric.argpartition#
- cunumeric.argpartition(a: ndarray, kth: Union[int, Sequence[int]], axis: int = -1, kind: SelectKind = 'introselect', order: Optional[Union[str, list[str]]] = None) ndarray #
Perform an indirect partition along the given axis.
- Parameters:
- Returns:
out – Array of indices that partitions a along the specified axis. It has the same shape as a.shape or is flattened in case of axis is None.
- Return type:
Notes
The current implementation falls back to cunumeric.argsort.
See also
- Availability:
Multiple GPUs, Multiple CPUs