def ce_pick

bendtorch@1.3.0.0 · main.bend:3217

import Base

3217# a label past the last class picks probability 0.0, clamped at 1e-12, as tensor-array's pick
3218def ce_pick(ok: Bool, +c: Nat, +off: U32, +idx: U32, loss: F32, z: Array<F32>) -> AS:
3219  match ok:
3220    case True{}:
3221      ce_pick3(c, off, loss, Array.get(F32, z, idx))
3222    case False{}:
3223      ce_bad(loss, ce_sm(c, off, z))