code: def __init__(self, input_size, hidden_size, depth):
      super(GraphGRU, self).__init__()
      self.hidden_size = hidden_size
      self.input_size = input_size
      self.depth = depth
  
      self.W_z = nn.Linear(input_size + hidden_size, hidden_size)
      self.W_r = nn.Linear(input_size, hidden_size, bias=False)
      self.U_r = nn.Linear(hidden_size, hidden_size)
      self.W_h = nn.Linear(input_size + hidden_size, hidden_size)
  
file_path: DIG/dig/ggraph/method/JTVAE/fast_jtnn/jtnn_enc.py
imports: import torch
  import torch.nn
  import torch.nn.functional
  from .mol_tree import Vocab, MolTree
  from .nnutils import create_var, index_select_ND
label_desc: Use image node_id 1 to create a new GraphGRU object from inherited base classes: nn with example: obj = GraphGRU(input_size, hidden_size, depth)