捷徑

Convolve

class torchaudio.transforms.Convolve(mode: str = 'full')[原始碼]

使用直接方法沿著輸入的最後一個維度進行卷積。請注意,與 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[原始碼]
參數:
  • x (torch.Tensor) – 第一個卷積運算元,形狀為 (…, N)

  • y (torch.Tensor) – 第二個卷積運算元,形狀為 (…, M) (前導維度必須可與 x 的維度廣播)。

返回:

卷積 xy 的結果,形狀為 (…, L),其中前導維度與 x 的維度相符,而 Lmode 決定。

返回類型:

torch.Tensor

文件

存取 PyTorch 的完整開發者文件

查看文件

教學

取得適用於初學者和進階開發者的深入教學

查看教學

資源

尋找開發資源並獲得問題解答

查看資源