DropBlock2d

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

請參閱 drop_block2d()。

forward(input: Tensor) → Tensor

引數:

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

返回:

經過 DropBlock 層後的張量。

返回型別:

張量