drop_block3d

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

实现了 DropBlock3D，出自 “DropBlock: A regularization method for convolutional networks” <https://arxiv.org/abs/1810.12890>。

参数:

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