PSRoIAlign
-
class torchvision.ops.PSRoIAlign(output_size: int, spatial_scale: float, sampling_ratio: int)[原始碼]
參見 ps_roi_align()。
-
forward(input: Tensor, rois: Tensor) → Tensor[原始碼]
定義每次呼叫時執行的計算。
應被所有子類覆蓋。
注意
儘管前向傳播的實現需要在該函式內定義，但之後應呼叫 Module 例項而不是直接呼叫此函式，因為前者會處理已註冊的鉤子，而後者則會默默忽略它們。
