code: def test_slice_axis(self):
      slice_layer = group_norm.GroupNorm(
          groups=5, create_scale=False, create_offset=False
      )
      axis_layer = group_norm.GroupNorm(
          groups=5, create_scale=False, create_offset=False
      )
  
      inputs = np.random.uniform(0, 10, [3, 4, 4, 5]).astype(np.float32)
      scale = np.random.normal(size=(5,), loc=1.0)
      offset = np.random.normal(size=(5,))
  
      slice_outputs = slice_layer(inputs, scale, offset)
      axis_outputs = axis_layer(inputs, scale, offset)
  
      self.assertAllClose(slice_outputs, axis_outputs)
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import group_norm
  from haiku._src import initializers
  from haiku._src import test_utils
  import jax.numpy
  import numpy
label_desc: Use image node_id 16 for calling the GroupNormTest obj's underlying member method code with example usage: obj.test_slice_axis() without return types