def nll_one

bendtorch@1.3.0.0 · main.bend:3284

import Base

3284# a label past the last class picks 0.0 and leaves its gradient row zero, as l_at and onehot do
3285def nll_one(ok: Bool, +idx: U32, +v: F32, acc: F32, lp: Array<F32>, g: Array<F32>) -> NS:
3286  match ok:
3287    case True{}:
3288      nll_hit(idx, v, g, acc, Array.get(F32, lp, idx))
3289    case False{}:
3290      NS{lp, g, (acc + 0.0 : F32)}