torch.nn.functional.feature_alpha_dropout

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

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

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

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

详情请参见 FeatureAlphaDropout。

Parameters

• p (float) – 通道被置零的dropout概率。默认值：0.5

• 训练 (bool) – 如果为True，则应用dropout。默认值：True

• inplace (bool) – 如果设置为 True，将在原地执行此操作。默认值：False

Return type

张量