~/jendhub

def Flat.rowsq.go

bend-tensors@0.0.0.2 · bend_tensors.bend:1989

import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensors
import Base

1989def Flat.rowsq.go(n: Nat, +i: U32, +mean: F32, r: Array<F32> & F32, +acc: F32) -> Array<F32> & F32:
1990  match n:
1991    case 0n:
1992      (a, x) = r
1993      (a, (acc + F32.sqdev(x, mean) : F32))
1994    case 1n+p:
1995      (a, x) = r
1996      Flat.rowsq.go(p, (i + 1 : U32), mean, Array.get(F32, a, (i + 1 : U32)), (acc + F32.sqdev(x, mean) : F32))

Uses (10)

Used by (1)