code: def __init__(self, node_dim=256, edge_dim=256):
      super().__init__()
      self.in_fc = nn.Linear(node_dim * 2 + edge_dim, node_dim)
      self.coef_fc = nn.Linear(node_dim, 1)
      self.out_fc = nn.Linear(node_dim, node_dim)
      self.relu = nn.ReLU()
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import math
  import paddle
  from paddle import nn
  import paddle.nn.functional
  from paddle import ParamAttr
label_desc: Use image node_id 1 to create a new GNNLayer object from inherited base classes: nn with example: obj = GNNLayer(node_dim, edge_dim)