code: def __init__(self, encoder, z_g_dim, z_n_dim):
      super(InfoG_enc, self).__init__()
      self.fc = nn.Linear(z_g_dim, z_n_dim)
      self.encoder = encoder
  
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 InfoG_enc object from inherited base classes: nn with example: obj = InfoG_enc(encoder, z_g_dim, z_n_dim)