espnet2.enh.layers.beamformer_th.get_rtf

source

espnet2.enh.layers.beamformer_th.get_rtf(psd_speech, psd_noise, mode='power', reference_vector: int | Tensor = 0, iterations: int = 3, diagonal_loading: bool = True, diag_eps: float = 1e-07)

Calculate the relative transfer function (RTF).

• Parameters:
  • psd_speech (torch.complex64) – speech covariance matrix (…, F, C, C)
  • psd_noise (torch.complex64) – noise covariance matrix (…, F, C, C)
  • mode (str) – one of (“power”, “evd”) “power”: power method “evd”: eigenvalue decomposition
  • reference_vector (torch.Tensor or int) – (…, C) or scalar
  • iterations (int) – number of iterations in power method
• Returns: (…, F, C)
• Return type: rtf (torch.complex64)