捷徑

stochastic_depth

torchvision.ops.stochastic_depth(input: Tensor, p: float, mode: str, training: bool = True) Tensor[原始碼]

實現來自「具有隨機深度的深度網路」的隨機深度,用於隨機丟棄殘差架構的殘差分支。

參數:
  • input (Tensor[N, ...]) – 輸入張量或任意維度,其中第一個是其批次,即具有 N 列的批次。

  • p (float) – 輸入歸零的機率。

  • mode (str) – "batch""row""batch" 隨機將整個輸入歸零,"row" 隨機將批次中選定的列歸零。

  • training – 如果為 True,則套用隨機深度。預設值:True

傳回:

隨機歸零的張量。

傳回類型:

Tensor[N, …]

文件

取得 PyTorch 的完整開發者文件

檢視文件

教學

取得適合初學者和進階開發者的深入教學

檢視教學

資源

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

檢視資源