class_docstr: A simple neural network for catch.
docstr: Unrolls more efficiently than dynamic_unroll.
docstr_tok: ["Unrolls","more","efficiently","than","dynamic_unroll","."]
code: def unroll(self, x, state):
      """Unrolls more efficiently than dynamic_unroll."""
      out, _ = hk.BatchApply(self)(x, None)
      return out, state
  
imports: import collections
  import dm_env
  import haiku
  import jax.nn
  import jax.numpy
label_desc: Use image node_id 4 for calling the CatchNet obj's underlying member method code with example usage: obj.unroll(x, state) and returns: out,  state