class_docstr: Head of GNN for node prediction.
code: def __init__(self, dim_in, dim_out):
      super().__init__()
      self.layer_post_mp = nn.Linear(dim_in, dim_out, bias=True)
  
imports: import torch.nn
  from torch_geometric.graphgym.register import register_head
label_desc: Use image node_id 1 to create a new ExampleNodeHead object from inherited base classes: nn with example: obj = ExampleNodeHead(dim_in, dim_out)