class_docstr: Implements test for Transforms that are performed for different devices
class_docstr_tok: ["Implements","test","for","Transforms","that","are","performed","for","different","devices"]
code: def test_PSD_with_mask(self):
      tensor = common_utils.get_whitenoise(
          sample_rate=8000, n_channels=4
      )
      spectrogram = common_utils.get_spectrogram(
          tensor, n_fft=400, hop_length=100
      )
      spectrogram = spectrogram.to(self.device)
      mask = torch.rand(spectrogram.shape[-2:], device=self.device)
      self._assert_consistency_complex(T.PSD(), spectrogram, mask)
  
imports: import torch
  import torchaudio.transforms
  from parameterized import parameterized
  from torchaudio_unittest import common_utils
  from torchaudio_unittest.common_utils import skipIfRocm, TestBaseMixin, torch_script
label_desc: Use image node_id 26 for calling the Transforms obj's underlying member method code with example usage: obj.test_PSD_with_mask() without return types