code: def __init__(self, graph_embed_func, node_hidden_size):
      super(AddEdge, self).__init__()
  
      self.graph_op = {"embed": graph_embed_func}
      self.add_edge = nn.Linear(
          graph_embed_func.graph_hidden_size + node_hidden_size, 1
      )
  
imports: import os
  import dgl
  import torch.nn
  import torch
  from functools import partial
  import torch.nn.functional
  from torch.distributions import Bernoulli
  from torch.distributions import Categorical
  import torch.utils.model_zoo
label_desc: Use image node_id 1 to create a new AddEdge object from inherited base classes: nn with example: obj = AddEdge(graph_embed_func, node_hidden_size)