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 層之後的張量。

回傳類型:

Tensor