• Zhaoheng Ni's avatar
    Add SoudenMVDR module (#2367) · aed5eb88
    Zhaoheng Ni authored
    Summary:
    Add a new design of MVDR module.
    The `SoudenMVDR` module supports the method proposed by [Souden et, al.](http://citeseerx.ist.psu.edu/viewdoc/download?doi=10.1.1.725.673&rep=rep1&type=pdf).
    The input arguments are:
    - multi-channel spectrum.
    - PSD matrix of target speech.
    - PSD matrix of noise.
    - reference channel in the microphone array.
    - diagonal_loading option to enable or disable diagonal loading in matrix inverse computation.
    - diag_eps for computing the inverse of the matrix.
    - eps for computing the beamforming weight.
    
    The output of the module is the single-channel complex-valued spectrum for the enhanced speech.
    
    Pull Request resolved: https://github.com/pytorch/audio/pull/2367
    
    Reviewed By: hwangjeff
    
    Differential Revision: D36198015
    
    Pulled By: nateanl
    
    fbshipit-source-id: 4027f4752a84aaef730ef3ea8c625e801cc35527
    aed5eb88
transforms.rst 2.95 KB