rms

rms(x)[source]

Calculate the Root Mean Square (RMS) of the input.

Parameters:

x (jnp.ndarray) – Input array.

Returns:

The RMS value.

Return type:

float or jnp.ndarray