class_docstr: Network for Atari.
code: def __init__(self, num_actions, use_resnet, use_lstm, name=None):
      super().__init__(name=name)
      self._num_actions = num_actions
      self._use_resnet = use_resnet
      self._use_lstm = use_lstm
      self._core = hk.ResetCore(hk.LSTM(256))
  
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 AtariNet object from inherited base classes: hk with example: obj = AtariNet(num_actions, use_resnet, use_lstm, name)