def Flat.map.go
import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensorsimport Base
1797def Flat.map.go(~f: F32 -> F32, n: Nat, +i: U32, a: Array<F32>) -> Array<F32>:
1798 match n:
1799 case 0n:
1800 a
1801 case 1n+p:
1802 Flat.map.go(~f, p, (i + 1 : U32), Flat.map.step(~f, i, Array.get(F32, a, i)))