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決定。- 返回型別:
