torch.nn.functional.feature_alpha_dropout#
- torch.nn.functional.feature_alpha_dropout(input, p=0.5, training=False, inplace=False)[source]#
随机掩盖整个通道(通道即特征图)。
例如,批次输入中第 个样本的第 个通道是输入张量中的一个张量 。与常规 Dropout 将激活值设为零不同,此操作会将激活值设为 SELU 激活函数的负饱和值。
在每次前向传播过程中,每个元素都会根据伯努利分布的样本以概率
p被独立掩盖。被掩盖的元素在每次前向传播时都是随机的,并经过缩放和平移以保持零均值和单位方差。详情请参阅
FeatureAlphaDropout。