torch.nn.functional.feature_alpha_dropout

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

隨機遮罩掉整個通道 (一個通道是一個特徵圖)。

例如，批次輸入中第 $j$ 個樣本的第 $i$ 個通道是一個張量 $\text{input}[i, j]$。與常規 Dropout 中將激活值設為零不同，這裡的激活值被設為 SELU 激活函數的負飽和值。

每個元素將在每次前向呼叫時以機率 p 獨立地被遮罩，使用來自 Bernoulli 分布的樣本。要被遮罩的元素在每次前向呼叫時都會隨機化，並且會被縮放和平移以維持零均值和單位變異數。

詳情請參閱 FeatureAlphaDropout。

參數

• p (float) – 通道被歸零的 dropout 機率。預設值：0.5

• training (bool) – 如果為 True 則套用 dropout。預設值：True

• inplace (bool) – 如果設定為 True，將會就地執行此操作。預設值：False

回傳類型

Tensor