class_docstr: See :func:`ps_roi_align`.
class_docstr_tok: ["See",":","func",":","`","ps_roi_align","`","."]
code: def __init__(
      self,
      output_size: int,
      spatial_scale: float,
      sampling_ratio: int,
  ):
      super().__init__()
      _log_api_usage_once(self)
      self.output_size = output_size
      self.spatial_scale = spatial_scale
      self.sampling_ratio = sampling_ratio
  
file_path: vision/torchvision/ops/ps_roi_align.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 1 to create a new PSRoIAlign object from inherited base classes: nn with example: obj = PSRoIAlign(output_size, spatial_scale, sampling_ratio)