stochastic_depth¶ torchvision.ops.stochastic_depth(input: Tensor, p: float, mode: str, training: bool = True) → Tensor[原始碼]¶ 實現 “Deep Networks with Stochastic Depth” 中的 Stochastic Depth，用於隨機丟棄殘差架構的殘差分支。 引數: input (Tensor[N, ...]) – 輸入張量或任意維度，第一個維度是批次，即具有 N 行的批次。 p (float) – 輸入被歸零的機率。 mode (str) – "batch" 或 "row"。 "batch" 會隨機歸零整個輸入，"row" 會隨機歸零批次中的選定行。 training – 如果為 True 則應用隨機深度。預設值：True 返回: 隨機歸零後的張量。 返回型別: Tensor[N, …]
