Pytorch Dropout函數

Dropout layers

class torch.nn.Dropout(p=0.5, inplace=False)

 

隨機將輸入張量中部分元素設置為0。對於每次前向調用，被置0的元素都是隨機的。

參數：

• p - 將元素置0的概率。默認值：0.5
• in-place - 若設置為True，會在原地執行操作。默認值：False

形狀：

• 輸入： 任意。輸入可以為任意形狀。
• 輸出： 相同。輸出和輸入形狀相同。

例子：

>>> m = nn.Dropout(p=0.2)
>>> input = autograd.Variable(torch.randn(20, 16))
>>> output = m(input)

>>> torch.nn.Dropout2d(p=0.5, inplace=False)

隨機將輸入張量中整個通道設置為0。對於每次前向調用，被置0的通道都是隨機的。