BendMlTensor.Mat.layernorm: def of bend-ml-tensor@0.1.2.0
Import: import bend-ml-tensor@0.1.2.0/main.bend as BendMlTensor
Lines 410-414 of bend-ml-tensor@0.1.2.0/main.bend: https://jend.leeeet.dev/read.txt?path=bend-ml-tensor@0.1.2.0/main.bend&start_line=410&end_line=414

import Base

# normalizes each row (d columns) and applies gamma and beta, vectors of d numbers
def Mat.layernorm(-n: Nat, d: Nat, eps: F32, g: Vec<d>, bt: Vec<d>, a: Mat<n, d>) -> Mat<n, d>:
  match g bt a:
    case Vec{gamma} Vec{beta} Mat{x}:
      Mat{rows.layernorm(x, F32.from_nat(d), eps, gamma, beta)}

Uses (6):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32.from_nat
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat
  https://jend.leeeet.dev/definition.txt?path=bend-ml-tensor@0.1.2.0/main.bend&name=Mat
  https://jend.leeeet.dev/definition.txt?path=bend-ml-tensor@0.1.2.0/main.bend&name=Vec
  https://jend.leeeet.dev/definition.txt?path=bend-ml-tensor@0.1.2.0/main.bend&name=rows.layernorm