code: def test_batchnorm_1d(self):
      x = objax.random.normal((64, 4, 16))
      bn = objax.nn.BatchNorm1D(4)
      # run batch norm in training mode
      yt = bn(x, training=True)
      self.assertEqual(yt.shape, x.shape)
      # run batch norm in eval mode
      ye = bn(x, training=False)
      self.assertEqual(ye.shape, x.shape)
  
imports: import os
  import unittest
  import jax.numpy
  import numpy
  import objax
label_desc: Use image node_id 2 for calling the TestBatchnorm obj's underlying member method code with example usage: obj.test_batchnorm_1d() without return types