tensorplay.nn.functional.rms_norm

tensorplay.nn.functional.rms_norm(input: TensorBase, normalized_shape, weight: TensorBase | None = None, eps: float | None = None) → TensorBase[source]

Apply Root Mean Square Layer Normalization.

Dispatches to the native fused kernel (single dispatch, CPU vectorized rows / CUDA block-per-row); falls back to the composite below under CompositeImplicitAutograd rms_norm.