code: def __init__(
      self,
      hidden_size,
      mode="lstm",
      return_sequences=False,
      bidirectional="False",
      num_layers=1,
  ):
      super().__init__()
      batch_first = True
      self.lstm = layer.CudnnRNN(
          hidden_size=hidden_size,
          batch_first=batch_first,
          rnn_mode=mode,
          return_sequences=return_sequences,
          num_layers=1,
          dropout=0.9,
          bidirectional=bidirectional,
      )
      self.l1 = layer.Linear(64)
      self.l2 = layer.Linear(2)
  
file_path: singa/examples/rnn/imdb_model.py
imports: from singa import autograd
  from singa import layer
  from singa import model
label_desc: Use image node_id 1 to create a new IMDBModel object from inherited base classes: model with example: obj = IMDBModel(hidden_size, mode, return_sequences, bidirectional, num_layers)