StochasticDepth

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

請參閱 stochastic_depth()。

forward(input: Tensor) → Tensor

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

應該由所有子類別覆寫。

注意

雖然前向傳遞的食譜需要在這個函式中定義，但之後應該呼叫 Module 實例，而不是這個，因為前者會處理執行已註冊的掛鉤，而後者會靜默地忽略它們。