~/jendhub

def cea_row0

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

import bend-ml-tensor-array@0.1.19.0/main.bend as BendMlTensorArray
import Base

465# a row's max starts from its first cell (maxl), then folds over the k + 1 cells from it
466def cea_row0(+mm1: Nat, +o: U32, +label: Nat, +nf: F32, +loss: F32, r: Array<F32> & F32) -> L2:
467  match r:
468    case Tuple{c, +x0}:
469      cea_row1(mm1, o, label, nf, loss, cea_max(mm1, o, x0, Array.get(F32, c, o)))

Uses (10)

Used by (1)