StochasticDepth

class torchvision.ops.StochasticDepth(p: float, mode: str)

參見 stochastic_depth()。

forward(input: Tensor) → Tensor

定義每次呼叫時執行的計算。

應由所有子類重寫。

注意

雖然前向傳播（forward pass）的實現需要在該函式中定義，但後續應呼叫 Module 例項而非此函式，因為前者會負責執行已註冊的鉤子（hooks），而後者會靜默忽略它們。