Pytorch中nn.Dropout2d的作用

首先,关于Dropout方法,这篇博文有详细的介绍。简单来说,

dropout方法有很多类型,图像处理中最常用的是Dropout2d,我从网上找了很多的中文资料,都没有让人满意的介绍,意外发现源代码dropout.py中的介绍还挺好的:

Randomly zero out entire channels:A channel is a 2D feature map.
Each channel will be zeroed out independently on every forward call with probability :attr:`p` using samples from a Bernoulli distribution

它是适用于有多个channel的二维输出的

05-11 20:44