def Flat.layer_norm.go
import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensorsimport Base
2034def Flat.layer_norm.go(n: Nat, +cols: Nat, +colsu: U32, +base: U32, +eps: F32, a: Array<F32>) -> Array<F32>:
2035 match n:
2036 case 0n:
2037 a
2038 case 1n+p:
2039 Flat.layer_norm.go(p, cols, colsu, (base + colsu : U32), eps, Flat.layer_norm.row(cols, base, eps, a))