drop_block2d

torchvision.ops.drop_block2d(input: Tensor, p: float, block_size: int, inplace: bool = False, eps: float = 1e-06, training: bool = True) → Tensor

實現 DropBlock2d，來自 “DropBlock: A regularization method for convolutional networks” <https://arxiv.org/abs/1810.12890>。

引數:

• input (Tensor[N, C, H, W]) – 輸入張量或 4 維張量，其中第一維是批次，即具有 N 行的批次。

• p (float) – 元素被丟棄的機率。

• block_size (int) – 要丟棄的塊的大小。

• inplace (bool) – 如果設定為 True，則就地執行此操作。預設值：False。

• eps (float) – 新增到分母中的值，用於數值穩定性。預設值：1e-6。

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

返回:

Dropblock 後隨機歸零的張量。

返回型別:

Tensor[N, C, H, W]