TensorPlay
API symbolsnn
Copy
View MarkdownDownload .md

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 — composed per the ATen rms_norm composite (fp32 compute for reduced dtypes).

Ask DeepWiki