3.1.21.1.2. unit_scaling.core.functional.rms

unit_scaling.core.functional.rms(x: Tensor, dims: Tuple[int, ...] | None = None, keepdim: bool = False, eps: float = 0.0) → Tensor

Compute the RMS \(\sqrt{\mathrm{mean}(x^2) + \epsilon}\) of a tensor.