code: def __init__(self, hidden_size):
      super(GatherUpdate, self).__init__()
      self.hidden_size = hidden_size
  
      self.W_o = nn.Linear(ATOM_FDIM + hidden_size, hidden_size)
  
imports: import dgl
  import dgl.function
  import rdkit.Chem
  import torch
  import torch.nn
  import torch.nn.functional
  from dgl import line_graph, mean_nodes
  from .chemutils import get_mol
label_desc: Use image node_id 1 to create a new GatherUpdate object from inherited base classes: nn with example: obj = GatherUpdate(hidden_size)