tensorplay.nn.functional.rms_norm

tensorplay.nn.functional.rms_norm(input: Tensor, normalized_shape, weight: Tensor | None = None, eps: float | None = None) → Tensor[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.