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，則此操作將在原地執行。預設值: False

返回型別

張量