快捷方式

drop_block3d

torchvision.ops.drop_block3d(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> 的 DropBlock3d。

参数:
  • input (Tensor[N, C, D, H, W]) – 输入张量,为 5 维,其中第一维是批次大小,即具有 N 行的批次。

  • p (float) – 元素被丢弃的概率。

  • block_size (int) – 要丢弃的块的大小。

  • inplace (bool) – 如果设置为 True,将原地执行此操作。默认值: False

  • eps (float) – 添加到分母以实现数值稳定性的值。默认值:1e-6。

  • training (bool) – 如果为 True 则应用 dropblock。默认值: True

返回值:

dropblock 后随机归零的张量。

返回类型:

Tensor[N, C, D, H, W]

文档

查阅 PyTorch 全面的开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源