code: def test_repeat(self, same_inputs, n, layer):
      rng_input, rng_params = random.split(random.PRNGKey(1), 2)
      x1 = jnp.cos(random.normal(rng_input, (2, 3)))
      x2 = None if same_inputs else random.normal(rng_input, (4, 3))
  
      self._test_repeat(x1, x2, layer, n, rng_params)
  
file_path: neural-tangents/tests/stax/combinators_test.py
imports: import random
  from absl.testing import absltest
  from jax import config
  from jax import random
  import jax.numpy
  from neural_tangents import stax
  from tests import test_utils
label_desc: Use image node_id 2 for calling the RepeatTest obj's underlying member method code with example usage: obj.test_repeat(same_inputs, n, layer) without return types