Function randomCategorical

randomCategorical(logits: MlxArray, key: MlxArray | null, s?: bigint): MlxArray
Parameters
- logits: MlxArray
- key: MlxArray | null
- s: bigint = gpuStream
Returns MlxArray
- Defined in packages/mlx/src/ops.ts:689