[ torch 参数更多 ]torch.nn.Dropout1d¶
torch.nn.Dropout1d¶
torch.nn.Dropout1d(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 | - | 表示在不更改变量的内存地址的情况下,直接修改变量的值,Paddle 无此参数,一般对网络训练结果影响不大,可直接删除。 |
- | axis | 指定对输入 Tensor 进行 Dropout 操作的轴,PyTorch 无此参数,Paddle 保持默认即可。 |
- | mode | 表示丢弃单元的方式,PyTorch 无此参数,Paddle 保持默认即可。 |