DropBlock3d

class torchvision.ops.DropBlock3d(p: float, block_size: int, inplace: bool = False, eps: float = 1e-06)

请参阅 drop_block3d()。

forward(input: Tensor) → Tensor

参数:

input (Tensor) – 将随机丢弃某些区域的输入特征图。

返回:

经过 DropBlock 层的张量。

返回类型:

张量