code: def testBatchNormNoScaleOrCenter(self):
      key = random.PRNGKey(0)
      axes = (0, 1, 2)
      init_fun, apply_fun = stax.BatchNorm(
          axis=axes, center=False, scale=False
      )
      input_shape = (4, 5, 6, 7)
      inputs = random_inputs(self.rng(), input_shape)
  
      out_shape, params = init_fun(key, input_shape)
      out = apply_fun(params, inputs)
      means = np.mean(out, axis=(0, 1, 2))
      std_devs = np.std(out, axis=(0, 1, 2))
      assert np.allclose(means, np.zeros_like(means), atol=1e-4)
      assert np.allclose(std_devs, np.ones_like(std_devs), atol=1e-4)
  
imports: from absl.testing import absltest
  import numpy
  from jax._src import test_util
  from jax import random
  from jax.example_libraries import stax
  from jax import dtypes
  from jax import config
label_desc: Use image node_id 15 for calling the StaxTest obj's underlying member method code with example usage: obj.testBatchNormNoScaleOrCenter() without return types