def Flat.rowscale.go
import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensorsimport Base
1948def Flat.rowscale.go(n: Nat, +i: U32, +k: F32, a: Array<F32>) -> Array<F32>:
1949 match n:
1950 case 0n:
1951 a
1952 case 1n+p:
1953 Flat.rowscale.go(p, (i + 1 : U32), k, Flat.scale.step(k, i, Array.get(F32, a, i)))