随机深度

torchvision.ops.stochastic_depth(input: Tensor, p: float, mode: str, training: bool = True) → Tensor

实现来自 “具有随机深度的深度网络” 的随机深度，用于随机丢弃残差架构的残差分支。

参数：

• input (Tensor[N, ...]) – 输入张量或任意维度，第一个维度是其批次，即具有 N 行的批次。

• p (float) – 输入被置零的概率。

• mode (str) – "batch" 或 "row"。 "batch" 随机将整个输入置零， "row" 将批次中随机选择的行置零。

• training – 如果为 True，则应用随机深度。默认值： True

返回：

随机置零的张量。

返回类型：

Tensor[N, …]