Missing import of torch.nn.functional

This line (https://gitlab.cern.ch/gnn4itkteam/commonframework/-/blob/dev/gnn4itk_cf/stages/graph_construction/models/metric_learning.py#L60) uses torch.nn.functional

Need to add

import torch.nn.functional as F