docstr: sdr_pit produces the same result as the reference implementation
code: def test_sdr_pit(self, batch_size, num_sources):
      """sdr_pit produces the same result as the reference implementation"""
      num_frames = 256
  
      estimation = torch.randn(batch_size, num_sources, num_frames)
      origin = torch.randn(batch_size, num_sources, num_frames)
  
      estimation -= estimation.mean(axis=2, keepdim=True)
      origin -= origin.mean(axis=2, keepdim=True)
  
      batch_sdr_ref = sdr_reference.batch_SDR_torch(estimation, origin)
      batch_sdr = metrics.sdr_pit(estimation, origin)
  
      self.assertEqual(batch_sdr, batch_sdr_ref)
  
file_path: audio/test/torchaudio_unittest/example/souce_sepration/metrics_test.py
imports: from itertools import product
  import torch
  from parameterized import parameterized
  from source_separation.utils import metrics
  from torch.testing._internal.common_utils import TestCase
  from .None import sdr_reference
label_desc: Use image node_id 2 for calling the TestSDR obj's underlying member method code with example usage: obj.test_sdr_pit(batch_size, num_sources) without return types