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

torch.nn.Dropout

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

paddle.nn.Dropout

paddle.nn.Dropout(p=0.5,
                  axis=None,
                  mode="upscale_in_train”,
                  name=None)

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

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | p | p | 表示丢弃概率。 | | inplace | - | 表示在不更改变量的内存地址的情况下,直接修改变量的值,PaddlePaddle 无此参数,一般对网络训练结果影响不大,可直接删除。 | | - | axis | 指定对输入 Tensor 进行 Dropout 操作的轴,PyTorch 无此参数,Paddle 保持默认即可。 | | - | mode | 表示丢弃单元的方式,PyTorch 无此参数,Paddle 保持默认即可。|