DropBlock3d

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

See drop_block3d().

forward(input: Tensor) → Tensor

Parameters:

input (Tensor) – Input feature map on which some areas will be randomly dropped.

Returns:

The tensor after DropBlock layer.

Return type:

Tensor