torch.nn.functional.dropout3d

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

隨機將整個通道歸零（通道是一個 3D 特徵圖）。

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

詳情請參閱 Dropout3d。

參數

• p (float) – 通道歸零的機率。預設值：0.5

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

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

回傳類型

Tensor