Function sdpa

sdpa(
    q: MlxArray,
    k: MlxArray,
    v: MlxArray,
    scale: number,
    maskMode: "" | "causal" | "array",
    maskArr?: MlxArray | null,
    s?: bigint,
): MlxArray
Parameters
- q: MlxArray
- k: MlxArray
- v: MlxArray
- scale: number
- maskMode: "" | "causal" | "array"
- maskArr: MlxArray | null = null
- s: bigint = gpuStream
Returns MlxArray
- Defined in packages/mlx/src/ops.ts:215