espnet2.enh.layers.beamformer.get_mwf_vector

source

espnet2.enh.layers.beamformer.get_mwf_vector(psd_s, psd_n, reference_vector: Tensor | int, diagonal_loading: bool = True, diag_eps: float = 1e-07, eps: float = 1e-08)

Return the MWF (Minimum Multi-channel Wiener Filter) vector:

h = (Npsd^-1 @ Spsd) @ u

• Parameters:
  • psd_s (torch.complex64/ComplexTensor) – speech covariance matrix (…, F, C, C)
  • psd_n (torch.complex64/ComplexTensor) – power-normalized observation covariance matrix (…, F, C, C)
  • reference_vector (torch.Tensor or int) – (…, C) or scalar
  • diagonal_loading (bool) – Whether to add a tiny term to the diagonal of psd_n
  • diag_eps (float)
  • eps (float)
• Returns: (…, F, C)
• Return type: beamform_vector (torch.complex64/ComplexTensor)