code: def compute_mat(self, query_emb, ref_emb):
      return torch.randn(query_emb.shape[0], ref_emb.shape[0])
  
file_path: pytorch-metric-learning/tests/distances/test_custom_check_shape.py
imports: import unittest
  import torch
  from pytorch_metric_learning.distances import BaseDistance, LpDistance
  from pytorch_metric_learning.losses import TripletMarginLoss
label_desc: Use image node_id 1 for calling the CustomDistance obj's underlying member method code with example usage: obj.compute_mat(query_emb, ref_emb) and returns: torch