def ce_row
import bend-ml-tensor-fork@0.1.6.0/main.bend as BendMlTensorForkimport Base
350def ce_row(+mm1: Nat, +mu: U32, +nf: F32, +i: U32, z: Array<F32>, t: Array<F32>, r: Array<U32> & U32) -> B3<F32, U32, F32>:
351 match r:
352 case Tuple{lab, +y}:
353 ce_put(i, lab, t, sr0(mm1, (i * mu : U32), U32.to_nat(y), nf, Array.get(F32, z, (i * mu : U32))))