評價此頁

torch.nn.functional.dropout3d#

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

隨機將整個通道(一個通道是3D特徵圖)置零。

例如,在批次輸入中,第 ii 個樣本的第 jj 個通道是輸入張量的一個3D張量 input[i,j]\text{input}[i, j]。在每次前向傳播呼叫時,每個通道都會根據伯努利分佈的樣本,以機率 p 獨立地置零。

有關詳細資訊,請參閱 Dropout3d

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

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

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

返回型別

張量