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 激活函数的负饱和值。

在每次前向调用中，每个元素都将使用伯努利分布（Bernoulli distribution）的样本以概率 p 被独立地遮蔽。每次前向调用时，要被遮蔽的元素都是随机的，并且会进行缩放和偏移以保持零均值和单位方差。

详见 FeatureAlphaDropout。

参数

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

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

• inplace (bool) – 如果设为 True，则此操作将就地进行。默认值：False

返回类型

Tensor