class_docstr: See :func:`roi_align`.
class_docstr_tok: ["See",":","func",":","`","roi_align","`","."]
code: def forward(
      self, input: Tensor, rois: Union[Tensor, List[Tensor]]
  ) -> Tensor:
      return roi_align(
          input,
          rois,
          self.output_size,
          self.spatial_scale,
          self.sampling_ratio,
          self.aligned,
      )
  
file_path: vision/torchvision/ops/roi_align.py
imports: from typing import List, Union
  import torch
  import torch._dynamo
  import torch.fx
  from torch import nn, Tensor
  from torch.jit.annotations import BroadcastingList2
  from torch.nn.modules.utils import _pair
  from torchvision.extension import _assert_has_ops, _has_ops
  from ..utils import _log_api_usage_once
  from ._utils import check_roi_boxes_shape, convert_boxes_to_roi_format
label_desc: Use image node_id 2 for calling the RoIAlign obj's underlying member method code with example usage: obj.forward(input, rois) and returns: roi_align