stochastic_depth¶ torchvision.ops.stochastic_depth(input: Tensor, p: float, mode: str, training: bool = True) → Tensor[source]¶ 实现来自 “Deep Networks with Stochastic Depth” 的 Stochastic Depth，用于随机丢弃残差架构的残差分支。 参数: input (Tensor[N, ...]) – 输入张量或任意维度，第一个维度是其批次，即具有 N 行的批次。 p (float) – 输入被置零的概率。 mode (str) – "batch" 或 "row"。 "batch" 会随机将整个输入置零，"row" 会随机选择批次中的行进行置零。 training – 如果为 True 则应用随机深度。 默认为：True 返回: 随机置零后的张量。 返回类型: Tensor[N, …]
