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 啟用函式的負飽和值。

使用來自伯努利分佈的樣本，每個元素將以 p 的機率在每次前向呼叫時獨立地被掩碼。被掩碼的元素將在每次前向呼叫時隨機生成，並進行縮放和移位，以保持零均值和單位方差。

有關詳細資訊，請參閱 FeatureAlphaDropout。

引數

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

• training (bool) – 如果為 True，則應用 dropout。預設為 True

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

返回型別

張量