DropBlock2d¶
-
class
torchvision.ops.DropBlock2d(p: float, block_size: int, inplace: bool = False, eps: float = 1e-06)[source]¶ See
drop_block2d().-
forward(input: torch.Tensor) → torch.Tensor[source]¶ - Parameters
input (Tensor) – Input feature map on which some areas will be randomly dropped.
- Returns
The tensor after DropBlock layer.
- Return type
Tensor
-
