Function fusedRespLogpMean

fusedRespLogpMean(
    model: RuntimeModel,
    hResp: MlxArray,
    targetsHost: Int32Array,
    chunkSize: number,
    sink: { dispose(): void }[],
    vocabBlock?: number,
    flash?: FlashCce | null,
): MlxArray
Parameters
- model: RuntimeModel
- hResp: MlxArray
- targetsHost: Int32Array
- chunkSize: number
- sink: { dispose(): void }[]
- vocabBlock: number = 0
- flash: FlashCce | null = null
Returns MlxArray
- Defined in packages/training/src/loss.ts:1089