def Flat.softmax.row
import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensorsimport Base
1964def Flat.softmax.row(+n: Nat, +base: U32, a: Array<F32>) -> Array<F32>:
1965 match n:
1966 case 0n:
1967 a
1968 case 1n+p:
1969 Flat.softmax.row2(1n+p, base, Flat.rowmax.go(p, base, Array.get(F32, a, base), (0.0 - 3.4028235e38 : F32)))