Dropout1d#

class torch.nn.Dropout1d(p=0.5, inplace=False)[原始碼]#

隨機將整個通道置零。

通道是 1D 特徵圖，例如，批次輸入中第 $i$ 個樣本的第 $j$ 個通道是一個 1D 張量 $\text{input}[i, j]$ 。

在每次前向傳播時，每個通道會以 p 的機率獨立地被清零，使用來自伯努利分佈的樣本。

通常輸入來自 nn.Conv1d 模組。

正如論文 Efficient Object Localization Using Convolutional Networks 中所述，如果特徵圖中的相鄰畫素強相關（這在早期卷積層中很常見），則 i.i.d. dropout 將不會正則化啟用，否則只會導致有效的學習率降低。

在這種情況下，nn.Dropout1d() 將有助於促進特徵圖之間的獨立性，並應被使用。

引數

形狀

示例

>>> m = nn.Dropout1d(p=0.2)
>>> input = torch.randn(20, 16, 32)
>>> output = m(input)

forward(input)[原始碼]#

執行前向傳播。

文件