[ torch 参数更多 ] torch.nn.Dropout2d

torch.nn.Dropout2d

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

paddle.nn.Dropout2D

paddle.nn.Dropout2D(p=0.5,
                    data_format='NCHW',
                    name=None)

PyTorch 相比 Paddle 支持更多其他参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
p p 表示丢弃概率。
inplace - 表示在不更改变量的内存地址的情况下,直接修改变量的值,Paddle 无此参数,一般对网络训练结果影响不大,可直接删除。
- data_format 指定对输入的数据格式,PyTorch 无此参数,Paddle 保持默认即可。