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 或 torch.Tensor) – 指定参考通道。如果 dtype 为
int，则表示参考通道索引。如果 dtype 为torch.Tensor，则其形状为 (…, channel)，其中channel维度是 one-hot 编码。diagonal_loading (bool, 可选) – 如果为
True，则启用对psd_n应用对角加载。(默认值:True)diag_eps (float, 可选) – 对角加载时乘以单位矩阵的系数。仅当
diagonal_loading设置为True时有效。(默认值:1e-7)
- 返回:
目标语音的估计复值 RTF。维度为 (…, freq, channel) 的张量。
- 返回类型:
- 使用
rtf_power的教程
