~/jendhub

def ce_sum

bend-ml-tensor-fork@0.1.6.0 · main.bend:373

import bend-ml-tensor-fork@0.1.6.0/main.bend as BendMlTensorFork
import Base

373# the loss over k terms, in row order from +0.0, as cea_rows adds them
374def ce_sum(k: Nat, +i: U32, +loss: F32, r: Array<F32> & F32) -> F32:
375  match k r:
376    case 0n Tuple{t, x}:
377      loss
378    case 1n+p Tuple{t, +x}:
379      ce_sum(p, (i + 1 : U32), (loss - x : F32), Array.get(F32, t, (i + 1 : U32)))

Uses (9)

Used by (1)