code: def __init__(self, input_dim, out_dim):
      super().__init__()
      self.block = nn.Sequential(
          nn.Linear(input_dim, out_dim),
          nn.ReLU(),
          nn.Linear(out_dim, out_dim),
          nn.ReLU(),
          nn.Linear(out_dim, out_dim),
          nn.ReLU(),
      )
      self.linear_shortcut = nn.Linear(input_dim, out_dim)
  
file_path: DIG/dig/sslgraph/method/contrastive/model/infograph.py
imports: import torch.nn
  from .contrastive import Contrastive
label_desc: Use image node_id 1 to create a new ProjHead object from inherited base classes: nn with example: obj = ProjHead(input_dim, out_dim)