快速鍵

drop_block2d

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

實作來自「DropBlock: convolutional networks 的正規化方法」的 DropBlock2d。

參數:
  • input (Tensor[N, C, H, W]) – 輸入張量或 4 維張量,其中第一個維度是其批次大小,即具有 N 列的批次。

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

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

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

  • eps (float) – 為數值穩定性添加到分母的值。預設值:1e-6。

  • training (bool) – 若為 True 則套用 dropblock。預設值:True

回傳:

經過 dropblock 後隨機歸零的張量。

回傳型別:

Tensor[N, C, H, W]

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

取得為初學者和進階開發者提供的深入教學

檢視教學

資源

尋找開發資源並獲得問題解答

檢視資源