code: def testSimpleCaseTensor(self):
      layer = axis_norm.LayerNorm(
          [1, 2], create_scale=False, create_offset=False
      )
  
      inputs = tf.ones([2, 3, 3, 5])
      scale = tf.constant(0.5, shape=(5,))
      offset = tf.constant(2.0, shape=(5,))
  
      outputs = layer(inputs, scale, offset).numpy()
      for x in np.nditer(outputs):
          self.assertEqual(x, 2.0)
  
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 5 for calling the LayerNormTest obj's underlying member method code with example usage: obj.testSimpleCaseTensor() without return types