class_docstr: A stateless agent interface.
class_docstr_tok: ["A","stateless","agent","interface","."]
docstr: Returns agent initial state.
docstr_tok: ["Returns","agent","initial","state","."]
code: def initial_state(self, batch_size: Optional[int]):
      """Returns agent initial state."""
      # We expect that generating the initial_state does not require parameters.
      return self._initial_state_apply_fn(None, batch_size)
  
file_path: dm-haiku/examples/impala/agent.py
imports: import collections
  import functools
  from typing import Any, Callable, Optional
  import dm_env
  import haiku
  from examples.impala import util
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 3 for calling the Agent obj's underlying member method code with example usage: obj.initial_state(batch_size) and returns: self