code: def __init__(self, hidden_size):
      nn.Module.__init__(self)
      self.hidden_size = hidden_size
  
      self.W_z = nn.Linear(2 * hidden_size, hidden_size)
      self.W_r = nn.Linear(hidden_size, hidden_size, bias=False)
      self.U_r = nn.Linear(hidden_size, hidden_size)
      self.W_h = nn.Linear(2 * hidden_size, hidden_size)
  
file_path: dgl/examples/pytorch/jtnn/jtnn/nnutils.py
imports: import os
  import dgl
  import torch
  import torch.nn
label_desc: Use image node_id 1 to create a new GRUUpdate object from inherited base classes: nn with example: obj = GRUUpdate(hidden_size)