code: def __init__(self, in_ft, out_ft):
      super(ProjHead, self).__init__()
      self.ffn = nn.Sequential(
          nn.Linear(in_ft, out_ft),
          nn.PReLU(),
          nn.Linear(out_ft, out_ft),
          nn.PReLU(),
          nn.Linear(out_ft, out_ft),
          nn.PReLU(),
      )
      self.linear_shortcut = nn.Linear(in_ft, out_ft)
  
file_path: DIG/dig/sslgraph/method/contrastive/model/mvgrl.py
imports: import torch.nn
  from .contrastive import Contrastive
  from dig.sslgraph.method.contrastive.views_fn import Diffusion, DiffusionWithSample
label_desc: Use image node_id 1 to create a new ProjHead object from inherited base classes: nn with example: obj = ProjHead(in_ft, out_ft)