[ 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 保持默认即可。|