快捷方式

DropBlock2d

class torchvision.ops.DropBlock2d(p: float, block_size: int, inplace: bool = False, eps: float = 1e-06)[source]

檢視 drop_block2d()

forward(input: Tensor) Tensor[source]
引數:

input (Tensor) – 輸入特徵圖,其中的某些區域將被隨機丟棄。

返回值:

DropBlock 層處理後的張量。

返回型別:

Tensor

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源