Shortcuts

DropBlock2d

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

See drop_block2d().

forward(input: Tensor) 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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources