code: def __init__(self, node_hidden_size):
      super(GraphEmbed, self).__init__()
  
      # Setting from the paper
      self.graph_hidden_size = 2 * node_hidden_size
  
      # Embed graphs
      self.node_gating = nn.Sequential(
          nn.Linear(node_hidden_size, 1), nn.Sigmoid()
      )
      self.node_to_graph = nn.Linear(
          node_hidden_size, self.graph_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 GraphEmbed object from inherited base classes: nn with example: obj = GraphEmbed(node_hidden_size)