code: def __init__(self, input_shape, n_actions, rnn_hidden_dim=64):
      super(RNNModel, self).__init__()
      self.rnn_hidden_dim = rnn_hidden_dim
  
      self.fc1 = nn.Linear(input_shape, rnn_hidden_dim)
      self.rnn = nn.GRUCell(rnn_hidden_dim, rnn_hidden_dim)
      self.fc2 = nn.Linear(rnn_hidden_dim, n_actions)
  
file_path: PARL/benchmark/torch/qmix/rnn_model.py
imports: import torch.nn
  import torch.nn.functional
  import parl
label_desc: Use image node_id 1 to create a new RNNModel object from inherited base classes: parl with example: obj = RNNModel(input_shape, n_actions, rnn_hidden_dim)