def Flat.argmax.row
import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensorsimport Base
2068def Flat.argmax.row(+n: Nat, +base: U32, a: Array<F32>) -> Array<F32> & (U32 & F32):
2069 match n:
2070 case 0n:
2071 (a, (0, 0.0))
2072 case 1n+p:
2073 Flat.rowarg.go(p, base, 0, Array.get(F32, a, base), (0, (0.0 - 3.4028235e38 : F32)))