docstr: Test ExponentialMovingAverage.
docstr_tok: ["Test","ExponentialMovingAverage","."]
code: def test_ExponentialMovingAverage(self):
      """Test ExponentialMovingAverage."""
      x1 = jn.array([[0, 1, 2]]) * 100
      x2 = jn.array([[-3, -4, 5]]) * 100
      init_value = 100
      shape = x1.shape
      ema = objax.nn.ExponentialMovingAverage(
          shape=shape, init_value=init_value, momentum=0.8
      )
  
      x_ema1 = ema(x1)
      x_ema2 = ema(x2)
  
      np.testing.assert_allclose(x_ema1, np.array([[80, 100, 120]]))
      np.testing.assert_allclose(x_ema2, np.array([[4, 0, 196]]))
  
file_path: objax/tests/nn_moving_average.py
imports: import unittest
  import jax.numpy
  import numpy
  import objax
label_desc: Use image node_id 2 for calling the TestMovingAverage obj's underlying member method code with example usage: obj.test_ExponentialMovingAverage() without return types