code: def test_exp_normalized(self):
      key = random.PRNGKey(0)
      x1 = random.normal(key, (2, 6, 7, 1))
      x2 = random.normal(key, (4, 6, 7, 1))
  
      for do_clip in [True, False]:
          for gamma in [1.0, 2.0, 0.5]:
              for get in ["nngp", "ntk"]:
                  with self.subTest(
                      do_clip=do_clip, gamma=gamma, get=get
                  ):
                      _, _, kernel_fn = stax.serial(
                          stax.Conv(1, (3, 3)),
                          stax.ExpNormalized(gamma, do_clip),
                          stax.Conv(1, (3, 3)),
                          stax.ExpNormalized(gamma, do_clip),
                          stax.GlobalAvgPool(),
                          stax.Dense(1),
                      )
                      k_12 = kernel_fn(x1, x2, get=get)
                      self.assertEqual(
                          k_12.shape, (x1.shape[0], x2.shape[0])
                      )
  
                      k_11 = kernel_fn(x1, None, get=get)
                      self.assertEqual(k_11.shape, (x1.shape[0],) * 2)
                      self.assertGreater(
                          jnp.min(jnp.linalg.eigvalsh(k_11)), 0
                      )
  
                      k_22 = kernel_fn(x2, None, get=get)
                      self.assertEqual(k_22.shape, (x2.shape[0],) * 2)
                      self.assertGreater(
                          jnp.min(jnp.linalg.eigvalsh(k_22)), 0
                      )
  
imports: import itertools
  import random
  from absl.testing import absltest
  from jax import config
  from jax import default_backend
  from jax import grad, jacfwd, jacrev, jit, jvp, value_and_grad
  from jax import random
  import jax.numpy
  import neural_tangents
  from neural_tangents import stax
  from neural_tangents._src.empirical import _DEFAULT_TESTING_NTK_IMPLEMENTATION
  from tests import test_utils
label_desc: Use image node_id 10 for calling the ActivationTest obj's underlying member method code with example usage: obj.test_exp_normalized() without return types