torchaudio.functional.apply_beamforming

torchaudio.functional.apply_beamforming(beamform_weights: Tensor, specgram: Tensor) → Tensor

将波束成形权重应用于多通道噪声频谱，以获得单通道增强频谱。

\[\hat{\textbf{S}}(f) = \textbf{w}_{\text{bf}}(f)^{\mathsf{H}} \textbf{Y}(f)\]

其中 \(\textbf{w}_{\text{bf}}(f)\) 是第 \(f\) 个频率的波束成形权重，\(\textbf{Y}\) 是第 \(f\) 个频率的通道频谱。

参数

• beamform_weights (Tensor) – 复值波束成形权重矩阵。张量维度为 (…, freq, channel)

• specgram (Tensor) – 多通道复值噪声频谱。张量维度为 (…, channel, freq, time)

返回

单通道复值增强频谱。

张量维度为 (…, freq, time)

返回类型

张量