class_docstr: See :func:`ps_roi_pool`.
class_docstr_tok: ["See",":","func",":","`","ps_roi_pool","`","."]
code: def forward(self, input: Tensor, rois: Tensor) -> Tensor:
      return ps_roi_pool(
          input, rois, self.output_size, self.spatial_scale
      )
  
file_path: vision/torchvision/ops/ps_roi_pool.py
imports: import torch
  import torch.fx
  from torch import nn, Tensor
  from torch.nn.modules.utils import _pair
  from torchvision.extension import _assert_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 PSRoIPool obj's underlying member method code with example usage: obj.forward(input, rois) and returns: ps_roi_pool