code: def get_pair_embeddings_with_ref(dtype, device):
      embeddings = torch.randn(
          4, 32, requires_grad=True, device=device, dtype=dtype
      )
      embeddings = torch.nn.functional.normalize(embeddings)
      labels = torch.LongTensor([0, 0, 1, 1])
  
      ref_emb = torch.randn(
          3, 32, requires_grad=True, device=device, dtype=dtype
      )
      ref_emb = torch.nn.functional.normalize(ref_emb)
      ref_labels = torch.LongTensor([0, 1, 2])
  
      pos_pairs = [
          (0, 0),
          (1, 0),
          (2, 1),
          (3, 1),
      ]
  
      neg_pairs = [
          (0, 1),
          (0, 2),
          (1, 1),
          (1, 2),
          (2, 0),
          (2, 2),
          (3, 0),
          (3, 2),
      ]
      return (
          embeddings,
          labels,
          ref_emb,
          ref_labels,
          pos_pairs,
          neg_pairs,
      )
  
label_desc: Use image node_id 2 for calling a global function with example usage: get_pair_embeddings_with_ref(dtype, device) and returns: embeddings,  labels,  ref_emb,  ref_labels,  pos_pairs,  neg_pairs