code: def __init__(self, input_dim, output_dim):
      super(PGNN_layer, self).__init__()
      self.input_dim = input_dim
  
      self.linear_hidden_u = nn.Linear(input_dim, output_dim)
      self.linear_hidden_v = nn.Linear(input_dim, output_dim)
      self.linear_out_position = nn.Linear(output_dim, 1)
      self.act = nn.ReLU()
  
file_path: dgl/examples/pytorch/P-GNN/model.py
imports: import dgl.function
  import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 1 to create a new PGNN_layer object from inherited base classes: nn with example: obj = PGNN_layer(input_dim, output_dim)