code: def forward(self, z_src, z_dst):
      h = self.lin_src(z_src) + self.lin_dst(z_dst)
      h = h.relu()
      return self.lin_final(h)
  
file_path: pytorch_geometric/examples/tgn.py
imports: import os.path
  import torch
  from sklearn.metrics import average_precision_score, roc_auc_score
  from torch.nn import Linear
  from torch_geometric.datasets import JODIEDataset
  from torch_geometric.loader import TemporalDataLoader
  from torch_geometric.nn import TGNMemory, TransformerConv
  from torch_geometric.nn.models.tgn import IdentityMessage, LastAggregator, LastNeighborLoader
label_desc: Use image node_id 2 for calling the LinkPredictor obj's underlying member method code with example usage: obj.forward(z_src, z_dst) and returns: self