class_docstr: Head of GNN, node prediction
class_docstr_tok: ["Head","of","GNN",",","node","prediction"]
code: def __init__(self, dim_in, dim_out):
      super(GNNNodeHead, self).__init__()
      self.layer_post_mp = MLP(
          dim_in, dim_out, num_layers=cfg.gnn.layers_post_mp, bias=True
      )
  
imports: import torch
  import torch.nn
  import graphgym.register
  from graphgym.config import cfg
  from graphgym.models.layer import MLP
  from graphgym.models.pooling import pooling_dict
label_desc: Use image node_id 1 to create a new GNNNodeHead object from inherited base classes: nn with example: obj = GNNNodeHead(dim_in, dim_out)