DropBlock2d

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

参见 drop_block2d()。

forward(input: Tensor) → Tensor

Parameters:

input (Tensor) – 输入特征图，将在其上随机丢弃一些区域。

Returns:

经过DropBlock层后的张量。

Return type:

张量