def cea_row3

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

import Base

450def cea_row3(+mm1: Nat, +o: U32, +label: Nat, +s: F32, +nf: F32, +loss: F32, q: AF) -> L2:
451  match q:
452    case AF{c, +p}:
453      L2{(loss - F32.log(F32.max(p, 0.000000000001)) : F32), cea_grad(mm1, o, 0n, label, s, nf, Array.get(F32, c, o))}