diff --git a/docs/source/functional.rst b/docs/source/functional.rst index 7d78b662e95..356da1a29ae 100644 --- a/docs/source/functional.rst +++ b/docs/source/functional.rst @@ -256,7 +256,6 @@ mvdr_weights_rtf .. autofunction:: mvdr_weights_rtf -compute_rtf_evd rtf_evd ------- diff --git a/test/torchaudio_unittest/functional/functional_impl.py b/test/torchaudio_unittest/functional/functional_impl.py index 1d5910f241b..d24a491ccd5 100644 --- a/test/torchaudio_unittest/functional/functional_impl.py +++ b/test/torchaudio_unittest/functional/functional_impl.py @@ -730,6 +730,11 @@ def _rtf_evd_numpy(self, psd): return rtf def test_rtf_evd(self): + """Verify ``F.rtf_evd`` method by the numpy implementation. + Given the multi-channel complex-valued spectrum, we compute the PSD matrix as the input, + ``F.rtf_evd`` outputs the relative transfer function (RTF) (Tensor of dimension `(..., freq, channel)`), + which should be identical to the output of ``rtf_evd_numpy``. + """ n_fft_bin = 10 channel = 4 specgram = np.random.random((n_fft_bin, channel)) + np.random.random((n_fft_bin, channel)) * 1j