code: def check_shapes(self, query_emb, ref_emb):
      if query_emb.ndim != 2 or (
          ref_emb is not None and ref_emb.ndim != 2
      ):
          raise ValueError(
              "embeddings must be a 2D tensor of shape (batch_size, embedding_size)"
          )
  
file_path: pytorch-metric-learning/src/pytorch_metric_learning/distances/base_distance.py
imports: import torch
  from ..utils.module_with_records import ModuleWithRecords
label_desc: Use image node_id 12 for calling the BaseDistance obj's underlying member method code with example usage: obj.check_shapes(query_emb, ref_emb) without return types