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