Convolve

class torchaudio.transforms.Convolve(mode: str = 'full')

使用直接方法对输入沿最后一个维度进行卷积。请注意，与实际应用有效交叉相关算子的 torch.nn.Conv1d 不同，本模块应用的是真正的 卷积 算子。

参数

mode (str, optional) –

必须是以下之一（“full”, “valid”, “same”）。

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

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

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

forward(x: Tensor, y: Tensor) → Tensor

参数

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

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

返回

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

返回类型

torch.Tensor