class_docstr: See :func:`stochastic_depth`.
code: def forward(self, input: Tensor) -> Tensor:
      return stochastic_depth(input, self.p, self.mode, self.training)
  
file_path: vision/torchvision/ops/stochastic_depth.py
imports: import torch
  import torch.fx
  from torch import nn, Tensor
  from ..utils import _log_api_usage_once
label_desc: Use image node_id 2 for calling the StochasticDepth obj's underlying member method code with example usage: obj.forward(input) and returns: stochastic_depth