class_docstr: Unrolls a recurrent module.
code: def __call__(self, x: jax.Array):
      initial_state = jax.tree_util.tree_map(
          lambda v: v.astype(x.dtype),
          self.wrapped.initial_state(batch_size=x.shape[0]),
      )
      x = jnp.expand_dims(x, axis=0)
      return self.unroller(self.wrapped, x, initial_state)
  
imports: from collections.abc import Sequence
  from typing import Any, Callable, NamedTuple
  import haiku
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 2 for calling the Recurrent obj's underlying member method code with example usage: obj.__call__(x) and returns: self