FFTConvolve

类 torchaudio.transforms.FFTConvolve(mode: 字符串 = 'full')

使用 FFT 对输入沿其最后一个维度进行卷积。对于最后一个维度较大的输入，此模块通常比 Convolve 快得多。请注意，与 torch.nn.Conv1d（实际应用的是有效互相关运算符）不同，此模块应用的是真正的卷积运算符。另请注意，此模块只能输出浮点张量（整数张量输入将被转换为浮点数）。

参数:

mode (字符串, 可选) –

必须是 (“full”, “valid”, “same”) 中的一个。

• ”full”: 返回完整的卷积结果，形状为 (…, N + M - 1)，其中 N 和 M 是两个输入的最后一个维度。(默认)

• ”valid”: 返回完整卷积结果中两个输入完全重叠的部分，形状为 (…, max(N, M) - min(N, M) + 1)。

• ”same”: 返回完整卷积结果的中心部分，形状为 (…, N)。

forward(x: 张量, y: 张量) → 张量

参数:

• x (torch.Tensor) – 第一个卷积操作数，形状为 (…, N)。

• y (torch.Tensor) – 第二个卷积操作数，形状为 (…, M)（前导维度必须能与 x 的维度广播兼容）。

返回:

x 和 y 的卷积结果，形状为 (…, L)，其中前导维度与 x 的维度匹配，L 由 mode 决定。

返回类型:

torch.Tensor