def cea_p

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

import Base

442# p[label] = e[label] / s, or 0.0 for a label past the row (pick's Nil case)
443def cea_p(inside: Bool, +o: U32, +label: Nat, +s: F32, c: Array<F32>) -> AF:
444  match inside:
445    case True{}:
446      cea_p2(s, Array.get(F32, c, (o + U32.from_nat(label) : U32)))
447    case False{}:
448      AF{c, 0.0}