stochastic_depth¶ torchvision.ops.stochastic_depth(input: Tensor, p: float, mode: str, training: bool = True) → Tensor[源]¶ 實現自 《具有隨機深度的深度網路》 的 Stochastic Depth 方法，用於隨機丟棄殘差架構中的殘差分支。 引數: input (Tensor[N, ...]) – 輸入張量，可以是任意維度，其中第一個維度表示批次，即具有 N 行的批次。 p (float) – 輸入被歸零的機率。 mode (str) – "batch" 或 "row"。"batch" 隨機將整個輸入歸零，"row" 隨機將批次中的選定行歸零。 training – 如果為 True，則應用隨機深度。預設值：True 返回值: 隨機歸零的張量。 返回型別: Tensor[N, …]
