cunumeric.argpartition#
- cunumeric.argpartition(a: ndarray, kth: int | Sequence[int], axis: int = -1, kind: SelectKind = 'introselect', order: str | list[str] | None = 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