~/jendhub

def ce_step

bend-ml-tensor-fork@0.1.6.0 · main.bend:355

import bend-ml-tensor-fork@0.1.6.0/main.bend as BendMlTensorFork
import Base

355def ce_step(+mm1: Nat, +mu: U32, +nf: F32, +i: U32, g: B3<F32, U32, F32>) -> B3<F32, U32, F32>:
356  match g:
357    case B3{z, lab, t}:
358      ce_row(mm1, mu, nf, i, z, t, Array.get(U32, lab, i))

Uses (6)

Used by (1)