class_docstr: Shallow torso for Atari, from the DQN paper.
code: def __init__(self, name=None):
      super().__init__(name=name)
  
imports: import collections
  import dm_env
  import haiku
  import jax.nn
  import jax.numpy
label_desc: Use image node_id 1 to create a new AtariShallowTorso object from inherited base classes: hk with example: obj = AtariShallowTorso(name)