class_docstr: Network for Atari.
code: def initial_state(self, batch_size):
      return self._core.initial_state(batch_size)
  
imports: import collections
  import dm_env
  import haiku
  import jax.nn
  import jax.numpy
label_desc: Use image node_id 2 for calling the AtariNet obj's underlying member method code with example usage: obj.initial_state(batch_size) and returns: self