optax.update_moment

Contents

optax.update_moment#

optax.update_moment(
updates,
moments,
decay,
order,
)[source]#

Compute the exponential moving average of the order-th moment.