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

torch.nn.Dropout ¶

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

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

         

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

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