快捷鍵

drop_block3d

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

實作來自 「DropBlock: 卷積網路的正規化方法」 的 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 的完整開發者文件

檢視文件

教學

取得針對初學者和進階開發者的深入教學

檢視教學

資源

尋找開發資源並獲得解答

檢視資源