code: def testDataFormatAgnosticVar(self):
      c_last_layer = axis_norm.LayerNorm(
          [1, 2], create_scale=True, create_offset=True
      )
      c_first_layer = axis_norm.LayerNorm(
          [2, 3],
          create_scale=True,
          create_offset=True,
          data_format="NCHW",
      )
  
      inputs = tf.random.uniform([3, 4, 4, 5], 0, 10)
  
      c_last_output = c_last_layer(inputs)
      inputs = tf.transpose(inputs, [0, 3, 1, 2])
      c_first_output = c_first_layer(inputs)
      c_first_output = tf.transpose(c_first_output, [0, 2, 3, 1])
  
      self.assertAllClose(c_last_output.numpy(), c_first_output.numpy())
  
imports: from absl.testing import parameterized
  import numpy
  from sonnet.src import axis_norm
  from sonnet.src import initializers
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 4 for calling the LayerNormTest obj's underlying member method code with example usage: obj.testDataFormatAgnosticVar() without return types