RoIAlign

class torchvision.ops.RoIAlign(output_size: None, spatial_scale: float, sampling_ratio: int, aligned: bool = False)

請參閱 roi_align()。

forward(input: Tensor, rois: Union[Tensor, List[Tensor]]) → Tensor

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

應由所有子類別覆寫。

注意

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