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]