评价此页

torch.nn.functional.feature_alpha_dropout#

torch.nn.functional.feature_alpha_dropout(input, p=0.5, training=False, inplace=False)[source]#

随机掩盖整个通道(通道即特征图)。

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

在每次前向传播过程中,每个元素都会根据伯努利分布的样本以概率 p 被独立掩盖。被掩盖的元素在每次前向传播时都是随机的,并经过缩放和平移以保持零均值和单位方差。

详情请参阅 FeatureAlphaDropout

参数:
  • p (float) – 通道被置零的丢弃概率。默认值:0.5

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

  • inplace (bool) – 如果设置为 True,则该操作将就地执行。默认值:False

返回类型:

张量