code: def setUp(self):
      super().setUp()
      self._batch_size = 32
      self._num_actions = 5
      self._input_shape = (84, 84, 4)
      self._batch_input_shape = (32, 84, 84, 4)
      self._rng = jax.random.PRNGKey(42)
      self._num_atoms = 21  # Rainbow/Quantile networks
      self._support = jax.numpy.linspace(-10, 10, self._num_atoms)
  
file_path: dopamine/tests/dopamine/jax/networks_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from dopamine.jax import networks
  from flax import linen
  import jax
  import numpy
label_desc: Use image node_id 1 for calling the NetworksTest obj's underlying member method code with example usage: obj.setUp() without return types