評價此頁

torch.nn.functional.dropout#

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

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

使用來自伯努利分佈的取樣。

有關詳細資訊,請參閱 Dropout

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

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

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

返回型別

張量