評價此頁

torch.nn.functional.dropout1d#

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

隨機將整個通道(通道是 1D 特徵圖)歸零。

例如,批次輸入中的第 ii 個樣本的第 jj 個通道是輸入張量的一個 1D 張量 input[i,j]\text{input}[i, j]。每個通道將在每次前向呼叫時以 p 的機率獨立地使用來自伯努利分佈的樣本進行歸零。

有關詳細資訊,請參閱 Dropout1d

引數
  • p (float) – 通道置零的機率。預設值:0.5

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

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

返回型別

張量