class_docstr: Simple convolutional stack commonly used for Atari.
class_docstr_tok: ["Simple","convolutional","stack","commonly","used","for","Atari","."]
code: def __init__(self):
      super().__init__(name="atari_torso")
      self._network = snt.Sequential(
          [
              snt.Conv2D(32, [8, 8], [4, 4]),
              tf.nn.relu,
              snt.Conv2D(64, [4, 4], [2, 2]),
              tf.nn.relu,
              snt.Conv2D(64, [3, 3], [1, 1]),
              tf.nn.relu,
              snt.Flatten(),
          ]
      )
  
imports: from typing import Optional, Tuple
  from acme.tf.networks import base
  from acme.tf.networks import duelling
  from acme.tf.networks import embedding
  from acme.tf.networks import policy_value
  from acme.tf.networks import recurrence
  from acme.tf.networks import vision
  from acme.wrappers import observation_action_reward
  import sonnet
  import tensorflow
label_desc: Use image node_id 1 to create a new AtariTorso object from inherited base classes: base with example: obj = AtariTorso()