def ce_rows

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

import Base

394def ce_rows(labels: List<&2, Nat>, +mm1: Nat, +mu: U32, +nf: F32, +i: U32, r: L2) -> L2:
395  match labels r:
396    case Nil{} L2{loss, c}:
397      L2{loss, c}
398    case Con{+y, t} L2{loss, c}:
399      ce_rows(t, mm1, mu, nf, (i + 1 : U32), ce_row(read_l(mm1, ((i * mu : U32) + U32.from_nat(mm1) : U32), Nil{}, Array.get(F32, c, ((i * mu : U32) + U32.from_nat(mm1) : U32))), y, nf, (i * mu : U32), loss))