Function topkAxis

topkAxis(a: MlxArray, k: number, axis: number, s?: bigint): MlxArray
mx.topk(a, k, axis) — the k largest values along axis (unsorted).
Parameters
- a: MlxArray
- k: number
- axis: number
- s: bigint = gpuStream
Returns MlxArray
- Defined in packages/mlx/src/ops.ts:1089