-
Notifications
You must be signed in to change notification settings - Fork 663
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add apply_beamforming to torchaudio.functional #2232
Conversation
test/torchaudio_unittest/functional/torchscript_consistency_impl.py
Outdated
Show resolved
Hide resolved
37ad9e6
to
d72cc84
Compare
@@ -582,6 +583,21 @@ def test_rnnt_loss_costs_and_gradients_random_data_with_numpy_fp32(self): | |||
ref_costs, ref_gradients = rnnt_utils.compute_with_numpy_transducer(data=data) | |||
self._test_costs_and_gradients(data=data, ref_costs=ref_costs, ref_gradients=ref_gradients) | |||
|
|||
def test_apply_beamforming(self): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Tests in functional_impl.py
implements test logics specific to each tested modules, so please add docstring of what is the expectation of this test.
test docstring should tell future maintainers (without context) what it is testing, so often it is the form of "(under this condition), given this input, this out comes should happen".
@nateanl has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
Summary: This PR adds ``apply_beamforming`` method to ``torchaudio.functional``. The method employs the beamforming weight to the multi-channel noisy spectrum to obtain the single-channel enhanced spectrum. The input arguments are the complex-valued beamforming weight Tensor and the multi-channel noisy spectrum. Pull Request resolved: pytorch#2232 Reviewed By: mthrok Differential Revision: D34474561 Pulled By: nateanl fbshipit-source-id: 8b1c3134b6e2d6e23cbedbb85022272af268b8d0
8fcf33c
to
ae90895
Compare
This pull request was exported from Phabricator. Differential Revision: D34474561 |
Summary: This PR adds ``apply_beamforming`` method to ``torchaudio.functional``. The method employs the beamforming weight to the multi-channel noisy spectrum to obtain the single-channel enhanced spectrum. The input arguments are the complex-valued beamforming weight Tensor and the multi-channel noisy spectrum. Pull Request resolved: pytorch#2232 Reviewed By: mthrok Differential Revision: D34474561 Pulled By: nateanl fbshipit-source-id: 2910251a8f111e65375dfb50495b6a415113f06d
This PR adds
apply_beamforming
method totorchaudio.functional
.The method employs the beamforming weight to the multi-channel noisy spectrum to obtain the single-channel enhanced spectrum.
The input arguments are the complex-valued beamforming weight Tensor and the multi-channel noisy spectrum.