快捷方式

卷積

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 的完整開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源