code: def __init__(
      self,
      state_dim,
      action_dim,
      num_hiddens,
      num_hidden_layers,
      num_gaussians,
  ) -> None:
      super().__init__()
      self.mdnrnn = MDNRNN(
          state_dim=state_dim,
          action_dim=action_dim,
          num_hiddens=num_hiddens,
          num_hidden_layers=num_hidden_layers,
          num_gaussians=num_gaussians,
      )
      self.state_dim = state_dim
      self.action_dim = action_dim
      self.num_hiddens = num_hiddens
      self.num_hidden_layers = num_hidden_layers
      self.num_gaussians = num_gaussians
  
file_path: ReAgent/reagent/models/world_model.py
imports: import torch
  from reagent.core import types
  from reagent.models.base import ModelBase
  from reagent.models.mdn_rnn import MDNRNN
label_desc: Use image node_id 1 to create a new MemoryNetwork object from inherited base classes: ModelBase with example: obj = MemoryNetwork(state_dim, action_dim, num_hiddens, num_hidden_layers, num_gaussians)