torchaudio.functional.fftconvolve¶
- torchaudio.functional.fftconvolve(x: Tensor, y: Tensor, mode: str = 'full') Tensor [源代码]¶
使用 FFT 沿着其最后一个维度对输入进行卷积。对于最后一个维度很大的输入,此函数通常比
convolve()
快得多。注意,与torch.nn.functional.conv1d()
不同,后者实际上应用了有效的互相关运算符,此函数应用了真正的 卷积 运算符。另请注意,此函数只能输出浮点张量(整数张量输入将被转换为浮点型)。- 参数:
x (torch.Tensor) – 第一个卷积操作数,形状为 (…, N)。
y (torch.Tensor) – 第二个卷积操作数,形状为 (…, M)(前导维度必须与
x
的维度广播兼容)。mode (str, 可选) –
必须是以下之一: (“full”, “valid”, “same”)。
”full”: 返回完整的卷积结果,形状为 (…, N + M - 1)。(默认)
”valid”: 返回对应于两个输入完全重叠部分的完整卷积结果的片段,形状为 (…, max(N, M) - min(N, M) + 1)。
”same”: 返回完整卷积结果的中心片段,形状为 (…, N)。
- 返回值:
对
x
和y
进行卷积的结果,形状为 (…, L),其中前导维度与x
的维度匹配,并且 L 由mode
决定。- 返回类型:
- 使用
fftconvolve
的教程 - 音频数据增强