espnet2.enh.layers.beamformer.get_power_spectral_density_matrix

source

espnet2.enh.layers.beamformer.get_power_spectral_density_matrix(xs, mask, normalization=True, reduction='mean', eps: float = 1e-15)

Return cross-channel power spectral density (PSD) matrix

• Parameters:
  • xs (torch.complex64/ComplexTensor) – (…, F, C, T)
  • reduction (str) – “mean” or “median”
  • mask (torch.Tensor) – (…, F, C, T)
  • normalization (bool)
  • eps (float)

Returns : psd (torch.complex64/ComplexTensor): (…, F, C, C)