optax.ScaleByRStdDevState#
- class optax.ScaleByRStdDevState(mu: optax.Updates, nu: optax.Updates)[source]#
State for centered exponential moving average of squares of updates.
State for centered exponential moving average of squares of updates.