class_docstr: A stateless agent interface.
class_docstr_tok: ["A","stateless","agent","interface","."]
docstr: Unroll the agent along trajectory.
docstr_tok: ["Unroll","the","agent","along","trajectory","."]
code: def unroll(
      self,
      params: hk.Params,
      trajectory: dm_env.TimeStep,
      state: Nest,
  ) -> AgentOutput:
      """Unroll the agent along trajectory."""
      net_out, _ = self._apply_fn(params, trajectory, state)
      return AgentOutput(
          net_out.policy_logits, net_out.value, action=[]
      )
  
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 5 for calling the Agent obj's underlying member method code with example usage: obj.unroll(params, trajectory, state) and returns: AgentOutput