Shortcuts

drop_block3d

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

实现来自“DropBlock: 卷积网络的正则化方法” 的DropBlock3d。

Parameters:
  • 输入 (张量[N, C, D, H, W]) – 输入张量或5维张量,第一个维度是批次大小,即一个包含N行的批次。

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

  • block_size (int) – 要删除的块的大小。

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

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

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

Returns:

经过dropblock后随机归零的张量。

Return type:

张量[N, C, D, H, W]