code: def __init__(self, graph_embed_func, node_hidden_size):
      super(AddNode, self).__init__()
  
      self.graph_op = {"embed": graph_embed_func}
  
      self.stop = 1
      self.add_node = nn.Linear(graph_embed_func.graph_hidden_size, 1)
  
      # If to add a node, initialize its hv
      self.node_type_embed = nn.Embedding(1, node_hidden_size)
      self.initialize_hv = nn.Linear(
          node_hidden_size + graph_embed_func.graph_hidden_size,
          node_hidden_size,
      )
  
      self.init_node_activation = torch.zeros(1, 2 * node_hidden_size)
  
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 AddNode object from inherited base classes: nn with example: obj = AddNode(graph_embed_func, node_hidden_size)