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