code: def initial_state(self, batch_size=1):
      if not self.use_lstm:
          return tuple()
      return tuple(
          torch.zeros(
              self.core.num_layers, batch_size, self.core.hidden_size
          )
          for _ in range(2)
      )
  
file_path: moolib/py/moolib/examples/atari/models.py
imports: import torch
  from torch import nn
  import torch.nn.functional
  from moolib.examples.common import nest
label_desc: Use image node_id 2 for calling the Net obj's underlying member method code with example usage: obj.initial_state(batch_size) and returns: tuple, tuple