espnet2.enh.layers.wpe.get_correlations

source

espnet2.enh.layers.wpe.get_correlations(Y: Tensor | ComplexTensor, inverse_power: Tensor, taps, delay) → Tuple[Tensor | ComplexTensor, Tensor | ComplexTensor]

Calculates weighted correlations of a window of length taps

• Parameters:
  • Y – Complex-valued STFT signal with shape (F, C, T)
  • inverse_power – Weighting factor with shape (F, T)
  • taps (int) – Lenghts of correlation window
  • delay (int) – Delay for the weighting factor
• Returns: Correlation matrix of shape (F, taps*C, taps*C) Correlation vector of shape (F, taps, C, C)