drop_block2d

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

实现来自“DropBlock: 一种用于卷积网络的正则化方法” <https://arxiv.org/abs/1810.12890>的 DropBlock2d。

参数：

• 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]