layerNorm static method

Tensor layerNorm(
  1. Tensor x,
  2. Tensor gamma,
  3. Tensor beta,
  4. double eps,
)

Implementation

static Tensor layerNorm(Tensor x, Tensor gamma, Tensor beta, double eps) {
  return Tensor._raw(
    engine.layernormForward(x._handle, gamma._handle, beta._handle, eps),
    x.shape,
  );
}