快捷方式

drop_block3d

torchvision.ops.drop_block3d(input: Tensor, p: float, block_size: int, inplace: bool = False, eps: float = 1e-06, training: bool = True) Tensor[原始碼]

實現了來自 “DropBlock: 一種用於卷積網路的正則化方法” <https://arxiv.org/abs/1810.12890> 的 DropBlock3d。

引數:
  • input (Tensor[N, C, D, H, W]) – 輸入張量,為 5 維,其中第一維是批次大小,即具有 N 行的批次。

  • p (float) – 元素被丟棄的機率。

  • block_size (int) – 要丟棄的塊的大小。

  • inplace (bool) – 如果設定為 True,將原地執行此操作。預設值: False

  • eps (float) – 新增到分母以實現數值穩定性的值。預設值:1e-6。

  • training (bool) – 如果為 True 則應用 dropblock。預設值: True

返回值:

dropblock 後隨機歸零的張量。

返回型別:

Tensor[N, C, D, H, W]

文件

查閱 PyTorch 全面的開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源