RoIPool

class torchvision.ops.RoIPool(output_size: None, spatial_scale: float)

請參閱 roi_pool()。

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

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

應由所有子類重寫。

注意

儘管正向傳播的實現需要在該函式內定義，但之後應呼叫 Module 例項而非直接呼叫此函式，因為前者會處理已註冊的鉤子，而後者則會默默忽略它們。