nn_build_rms_norm: Build RMS normalization forward pass

View source: R/nn_layers.R

nn_build_rms_normR Documentation

Build RMS normalization forward pass

Description

RMS-normalize, then scale by gamma and shift by beta. No batch statistics are involved, so training and inference build the same graph.

Usage

nn_build_rms_norm(ctx, input_tensor, layer)

Value

A ggml_tensor.


ggmlR documentation built on Oct. 23, 2026, 1:07 a.m.