code: def __init__(
      self,
      units: int,
      num_layers: int,
      adj_matrix: Tensor,
      use_mlp: bool = True,
      **kwargs,
  ):
      super().__init__(**kwargs)
  
      self.units = units
      self.num_layers = num_layers
      self.adj_matrix = adj_matrix
      self.use_mlp = use_mlp
  
      if self.use_mlp:
          with self.name_scope():
              self.gnn_layer = mx.gluon.nn.Dense(
                  units=self.units, flatten=False
              )
  
imports: import mxnet
  from gluonts.mx import Tensor
label_desc: Use image node_id 1 to create a new GNN object from inherited base classes: mx.gluon with example: obj = GNN(units, num_layers, adj_matrix, use_mlp)