class_docstr: Maintains an exponential moving average.
  
  This uses the Adam debiasing procedure.
  See https://arxiv.org/pdf/1412.6980.pdf for details.
class_docstr_tok: ["Maintains","an","exponential","moving","average",".","This","uses","the","Adam","debiasing","procedure",".","See","https",":","\/\/arxiv.org\/pdf\/1412.6980.pdf","for","details","."]
code: def average(self):
      return hk.get_state("average")
  
imports: import re
  from typing import Optional, Union
  import warnings
  from haiku._src import base
  from haiku._src import data_structures
  from haiku._src import initializers
  from haiku._src import module
  import jax
  import jax.numpy
label_desc: Use image node_id 4 for calling the ExponentialMovingAverage obj's underlying member method code with example usage: obj.average() and returns: hk