快捷方式

卷积

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

使用直接方法沿其最后一个维度对输入进行卷积。请注意,与 torch.nn.Conv1d 实际上应用有效互相关运算符相反,此模块应用真正的 卷积 运算符。

This feature supports the following devices: CPU, CUDA This API supports the following properties: Autograd, TorchScript
参数:

mode (str, 可选) –

必须是 (“full”, “valid”, “same”) 之一。

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

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

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

forward(x: Tensor, y: Tensor) Tensor[source]
参数:
  • x (torch.Tensor) – 第一个卷积操作数,形状为 (…, N)

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

返回值:

xy 卷积的结果,形状为 (…, L),其中前导维度与 x 的前导维度匹配,而 Lmode 决定。

返回类型:

torch.Tensor

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源