Python PyTorch conv1d用法及代碼示例

本文簡要介紹python語言中 torch.nn.functional.conv1d 的用法。

用法:

torch.nn.functional.conv1d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1) → Tensor

參數：

• input-形狀的輸入張量(\text{minibatch} , \text{in\_channels} , iW)

• weight-形狀過濾器(\text{out\_channels} , \frac{\text{in\_channels}}{\text{groups}} , kW)

• bias-形狀 (\text{out\_channels}) 的可選偏差。默認值：None

• stride-卷積核的步幅。可以是單個數字或單元素元組 (sW,) 。默認值：1

• padding-

  輸入兩側的隱式填充。可以是字符串 {‘valid’, ‘same’}、單個數字或單元素元組 (padW,) 。默認值：0 padding='valid' 與無填充相同。 padding='same' 填充輸入，以便輸出具有與輸入相同的形狀。但是，此模式不支持除 1 之外的任何步幅值。

  警告

  對於 padding='same' ，如果 weight 是偶數長度並且 dilation 在任何維度上都是奇數，則內部可能需要完整的 pad() 操作。降低性能。

• dilation-內核元素之間的間距。可以是單個數字或單元素元組 (dW,) 。默認值：1

• groups-將輸入分成組，\text{in\_channels} 應該可以被組數整除。默認值：1

對由多個輸入平麵組成的輸入信號應用一維卷積。

該運算符支持 TensorFloat32。

有關詳細信息和輸出形狀，請參見 Conv1d 。

注意

在某些情況下，當在 CUDA 設備上給定張量並使用 CuDNN 時，此運算符可能會選擇非確定性算法來提高性能。如果這是不可取的，您可以嘗試通過設置 torch.backends.cudnn.deterministic = True 來使操作具有確定性(可能以性能為代價)。有關詳細信息，請參閱重現性。

例子：

>>> inputs = torch.randn(33, 16, 30)
>>> filters = torch.randn(20, 16, 5)
>>> F.conv1d(inputs, filters)