def cea_rows

bend-ml-tensor-array@0.1.19.0 · main.bend:471

import Base

471# one row per label, in order, as ce_rows
472def cea_rows(labels: List<&2, Nat>, +mm1: Nat, +mu: U32, +nf: F32, +i: U32, r: L2) -> L2:
473  match labels r:
474    case Nil{} L2{loss, c}:
475      L2{loss, c}
476    case Con{+y, t} L2{+loss, c}:
477      cea_rows(t, mm1, mu, nf, (i + 1 : U32), cea_row0(mm1, (i * mu : U32), y, nf, loss, Array.get(F32, c, (i * mu : U32))))