code: def test_computes(self):
      hidden_size = (4, 4)
      output_size = 6
      model = atari_cnn.AtariCnn(
          hidden_sizes=hidden_size, output_size=output_size
      )
      B, T, OBS = 2, 2, (28, 28, 3)  # pylint: disable=invalid-name
      input_signature = ShapeDtype((1, 1) + OBS)
      _, _ = model.init(input_signature)
      x = np.arange(
          B * (T + 1) * functools.reduce(op.mul, OBS)
      ).reshape(B, T + 1, *OBS)
      y = model(x)
      self.assertEqual((B, T + 1, output_size), y.shape)
  
file_path: trax/trax/models/atari_cnn_test.py
imports: import functools
  import operator
  import numpy
  from tensorflow import test
  from trax.models import atari_cnn
  from trax.shapes import ShapeDtype
label_desc: Use image node_id 1 for calling the AtariCnnTest obj's underlying member method code with example usage: obj.test_computes() without return types