Function choice

choice(
    key: Array,
    a: ArrayLike,
    __namedParameters?: {
        axis?: number;
        p?: ArrayLike;
        replace?: boolean;
        shape?: number[];
    },
): Array
Sample from a population with optional replacement and optional probabilities.

This implements the common JAX-compatible cases: integer populations and array populations along axis. Probabilities p, if provided, are sampled via categorical(log(p)).
Parameters
- key: Array
- a: ArrayLike
- __namedParameters: { axis?: number; p?: ArrayLike; replace?: boolean; shape?: number[] } = {}
Returns Array
- Defined in src/library/random.ts:280