3.1.20.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[source]: Compute the RMS $\sqrt{mean (x^{2}) + ϵ}$ of a tensor.