def Flat.rowexp.step
import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensorsimport Base
1934# exp(x - mx) written back, the row's sum answered
1935def Flat.rowexp.step(+i: U32, +mx: F32, r: Array<F32> & F32, +acc: F32) -> Array<F32> & F32:
1936 (a, x) = r
1937 +y = F32.exp((x - mx : F32))
1938 (Array.set(F32, a, i, y), (acc + y : F32))