torch.nn.functional.dropout

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

在訓練期間，以機率 p 隨機將輸入張量 (tensor) 的某些元素歸零。

使用來自 Bernoulli 分佈的樣本。

詳情請參閱 Dropout。

參數

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

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

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

回傳類型

Tensor (張量)