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, …]