qfeval_functions.functions.rms

rms(x, dim=(), keepdim=False)

Returns the root mean square of each row of the input tensor in the given dimension dim. If dim is a list of dimensions, reduce over all of them.

Parameters:

• x (Tensor) – The input tensor.

• dim (Union[int, Tuple[int, ...]]) – The dimension or dimensions to reduce.

• keepdim (bool) – Whether the output tensor has dim retained or not.

Returns:

The output tensor.

Return type:

Tensor