ltx23_rms_norm

RMS normalization

Description

Variance is computed in float32; the result is cast back to the input dtype (or the weight dtype when elementwise affine).

Usage

ltx23_rms_norm(dim, eps = 1e-06, elementwise_affine = TRUE)

Arguments

  • dim: Integer. Normalized dimension size.
  • eps: Numeric. Stability epsilon.
  • elementwise_affine: Logical. Learn a scale weight.