torchaudio.functional.rtf_power¶
- torchaudio.functional.rtf_power(psd_s: Tensor, psd_n: Tensor, reference_channel: Union[int, Tensor], n_iter: int = 3, diagonal_loading: bool = True, diag_eps: float = 1e-07) Tensor[source]¶
使用幂法估计相对传递函数 (RTF) 或转向矢量。
- 参数:
psd_s (torch.Tensor) – 目标语音的复数功率谱密度 (PSD) 矩阵。具有尺寸为 (…, freq, channel, channel) 的张量。
psd_n (torch.Tensor) – 噪声的复数功率谱密度 (PSD) 矩阵。具有尺寸为 (…, freq, channel, channel) 的张量。
reference_channel (int or torch.Tensor) – 指定参考通道。如果数据类型为
int，则表示参考通道索引。如果数据类型为torch.Tensor，则其形状为 (…, channel)，其中channel维度为独热编码。diagonal_loading (bool, 可选) – 如果为
True，则启用对psd_n应用对角加载。（默认值：True）diag_eps (float, 可选) – 用于对角加载的乘以单位矩阵的系数。它仅在
diagonal_loading设置为True时有效。（默认值：1e-7）
- 返回:
估计的目标语音的复数 RTF。维度为 (…, freq, channel) 的张量。
- 返回类型:
- 使用
rtf_power的教程
