class_docstr: GRU class.
class_docstr_tok: ["GRU","class","."]
code: def __init__(self, name, input_dim, hidden_dim):
      self.name = "/".join([name, "gru"])
      self.input_dim = input_dim
      self.hidden_dim = hidden_dim
      self.w_matrix = None
      self.U = None
      self.bias = None
  
file_path: nni/examples/trials/ga_squad/rnn.py
imports: import tensorflow
  from tensorflow.python.ops.rnn_cell_impl import RNNCell
label_desc: Use image node_id 1 to create a new GRU object  with example: obj = GRU(name, input_dim, hidden_dim)