torch.nn.functional.dropout

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

在訓練期間，根據機率 p 隨機將輸入張量的一些元素歸零。

使用伯努利分佈取樣。

詳見 Dropout。

引數

• p (float) – 將元素歸零的機率。預設值：0.5

• training (bool) – 若為 True 則應用 dropout。預設值：True

• inplace (bool) – 若設為 True，則原地執行此操作。預設值：False

返回型別

張量