code: def test_data_format_agnostic_var(self):
      c_last_layer = group_norm.GroupNorm(
          groups=5, create_scale=True, create_offset=True
      )
      c_first_layer = group_norm.GroupNorm(
          groups=5,
          create_scale=True,
          create_offset=True,
          data_format="NCHW",
      )
  
      inputs = np.random.uniform(0, 10, [3, 4, 4, 10]).astype(
          np.float32
      )
  
      c_last_output = c_last_layer(inputs)
      inputs = jnp.transpose(inputs, [0, 3, 1, 2])
      c_first_output = c_first_layer(inputs)
      c_first_output = jnp.transpose(c_first_output, [0, 2, 3, 1])
  
      self.assertAllClose(c_last_output, c_first_output)
  
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 4 for calling the GroupNormTest obj's underlying member method code with example usage: obj.test_data_format_agnostic_var() without return types