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