code: def testSimpleTraining(self):
      layer = batch_norm.BaseBatchNorm(
          moving_mean=TestMetric(),
          moving_variance=TestMetric(),
          create_scale=False,
          create_offset=False,
      )
  
      inputs = tf.ones([2, 3, 3, 5])
      scale = tf.constant(0.5, shape=(5,))
      offset = tf.constant(2.0, shape=(5,))
  
      outputs = layer(inputs, True, scale=scale, offset=offset).numpy()
      self.assertAllEqual(outputs, tf.fill(inputs.shape, 2.0))
      self.assertEqual((0, 1, 2), layer._axis)
  
imports: import itertools
  from absl.testing import parameterized
  import numpy
  from sonnet.src import batch_norm
  from sonnet.src import initializers
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 1 for calling the BaseBatchNormTest obj's underlying member method code with example usage: obj.testSimpleTraining() without return types