torch.nn.functional.feature_alpha_dropout

torch.nn.functional.feature_alpha_dropout(input, p=0.5, training=False, inplace=False)

随机屏蔽整个通道（通道是一个特征图）。

例如，批输入中第 $j$ 个通道的第 $i$ 个样本是输入张量 $\text{input}[i, j]$ 。与常规 Dropout 将激活设置为 0 不同，激活被设置为 SELU 激活函数的负饱和值。

每个元素将在每次前向调用中以概率 p 独立进行掩码，使用伯努利分布的样本。要掩码的元素将在每次前向调用中随机化，并缩放和偏移以保持零均值和单位方差。

详细内容请见 FeatureAlphaDropout 。

参数:

• p（浮点数）- 要置零的通道的 dropout 概率。默认：0.5

• training (布尔值) – 如果是 True 则应用 dropout。默认值： True

• inplace（布尔值）- 如果设置为 True ，将在此处执行此操作。默认： False

返回类型:

张量