torch.nn.functional.feature_alpha_dropout

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

随机掩码掉整个通道（通道是特征图）。

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

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

有关详细信息，请参阅 FeatureAlphaDropout。

参数

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

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

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

返回类型

张量