def Flat.softmax.row2
import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensorsimport Base
1960def Flat.softmax.row2(+n: Nat, +base: U32, r: Array<F32> & F32) -> Array<F32>:
1961 (a, mx) = r
1962 Flat.softmax.row3(n, base, Flat.rowexp.go(n, base, mx, (a, 0.0)))