code: def test_network_outputs_correct_shapes(self):
      network_def = continuous_networks.SACEncoderNetwork()
      params = network_def.init(self.key, self.example_input)
  
      output = network_def.apply(params, self.example_input)
  
      self.assertIsInstance(
          output, continuous_networks.SACEncoderOutputs
      )
      with self.subTest("critic"):
          self.assertEqual(output.critic_z.shape, (50,))
      with self.subTest("actor"):
          self.assertEqual(output.actor_z.shape, (50,))
  
file_path: dopamine/tests/dopamine/labs/sac_from_pixels/continuous_networks_test.py
imports: from absl.testing import absltest
  from dopamine.labs.sac_from_pixels import continuous_networks
  import jax
  from jax import numpy
label_desc: Use image node_id 2 for calling the SACEncoderNetworkTest obj's underlying member method code with example usage: obj.test_network_outputs_correct_shapes() without return types