class_docstr: Recurrent Neural Network (RNN) block.
docstr: Initialize hidden state for input batch of size ``batch_size``.
code: def init_state(self, batch_size):
      """Initialize hidden state for input batch of size ``batch_size``."""
      self.state = StateVar(jn.zeros((batch_size, self.nstate)))
  
file_path: objax/objax/zoo/rnn.py
imports: from typing import Callable
  import jax.numpy
  from objax import Module
  from objax.nn import Linear
  from objax.nn.init import kaiming_normal
  from objax.typing import JaxArray
  from objax.variable import TrainVar, StateVar
label_desc: Use image node_id 2 for calling the RNN obj's underlying member method code with example usage: obj.init_state(batch_size) without return types