Root Mean Square normalization (RMS norm).
The normalization is with respect to the last axis of the input x.
- 参数:
x (array) -- Input array.
weight (array, optional) -- A multiplicative weight to scale the result by.
The weight should be one-dimensional with the same size
as the last axis of x. If set to None then no scaling happens.
eps (float) -- A small additive constant for numerical stability.
- 返回:
The output array.
- 返回类型:
array
