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的通道都是随机的。