alpha_dropout

paddle.nn.functional.alpha_dropout ( x, p=0.5, training=True, name=None ) [源代码]

alpha_dropout是一种具有自归一化性质的dropout。均值为0，方差为1的输入，经过alpha_dropout计算之后，输出的均值和方差与输入保持一致。alpha_dropout通常与SELU激活函数组合使用。

参数

• x (Tensor): 输入的多维 Tensor ，数据类型为：float32、float64。

• p (float): 将输入节点置0的概率，即丢弃概率。默认: 0.5。

• training (bool): 标记是否为训练阶段。 默认: True。

• name (str，可选): 操作的名称(可选，默认值为None）。更多信息请参见 Name 。

返回

经过alpha_dropout之后的结果，与输入x形状相同的 Tensor 。

代码示例

import paddle
import numpy as np
x = np.array([[-1, 1], [-1, 1]]).astype('float32')
x = paddle.to_tensor(x)
y_train = paddle.nn.functional.alpha_dropout(x, 0.5)
y_test = paddle.nn.functional.alpha_dropout(x, 0.5, training=False)
print(x)
print(y_train)
# [[-0.10721093, 1.6655989 ], [-0.7791938, -0.7791938]] (randomly)
print(y_test)