code: def test_psd_with_mask(self, multi_mask):
      transform = T.PSD(multi_mask=multi_mask)
      waveform = get_whitenoise(
          sample_rate=8000, duration=0.05, n_channels=2
      )
      spectrogram = get_spectrogram(waveform, n_fft=400)
      if multi_mask:
          mask = torch.rand(spectrogram.shape[-3:])
      else:
          mask = torch.rand(spectrogram.shape[-2:])
  
      self.assert_grad(transform, [spectrogram, mask])
  
imports: import unittest
  from typing import List
  import torch
  import torchaudio.transforms
  from parameterized import parameterized
  from torch.autograd import gradcheck, gradgradcheck
  from torchaudio_unittest.common_utils import get_spectrogram, get_whitenoise, nested_params, rnnt_utils, TestBaseMixin
label_desc: Use image node_id 22 for calling the AutogradTestMixin obj's underlying member method code with example usage: obj.test_psd_with_mask(multi_mask) without return types