code: def test_simple_case_tensor(self):
      layer = layer_norm.LayerNorm(
          [1, 2], create_scale=False, create_offset=False, param_axis=-1
      )
  
      inputs = np.ones([2, 3, 3, 5])
      scale = np.full((5,), 0.5)
      offset = np.full((5,), 2.0)
  
      outputs = layer(inputs, scale, offset)
      for x in np.nditer(outputs):
          self.assertEqual(x, 2.0)
  
imports: import functools
  import itertools
  from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import initializers
  from haiku._src import layer_norm
  from haiku._src import test_utils
  from haiku._src import transform
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 5 for calling the LayerNormTest obj's underlying member method code with example usage: obj.test_simple_case_tensor() without return types