optax.update_moment# optax.update_moment( updates, moments, decay, order, )[source]# Compute the exponential moving average of the order-th moment.