[ 仅参数默认值不一致 ]torch.alpha_dropout

torch.alpha_dropout

torch.alpha_dropout(input, p=0.5, train=False)

paddle.nn.functional.alpha_dropout

paddle.nn.functional.alpha_dropout(x, p=0.5, training=True, name=None)

其中 PyTorch 和 Paddle 功能一致,仅参数默认值不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
input x 输入的多维 Tensor,仅参数名不一致。
p p 将输入节点置 0 的概率,即丢弃概率。
train training 标记是否为训练阶段。PyTorch 默认值为 False,Paddle 默认值为 True。Paddle 需设置为与 PyTorch 一致。