Thank you @ptrblck ! That responds my question since I was interested in learning about how the ROI Align algorith was implemented since I was not able to understand just reading the Mask R CNN Paper.
If you don’t mind I’d like to ask another question now. Is there some documentation explaining how this cpp code is loaded into the python side? When I call torch.ops.torchvision.roi_align, how does that load the cpp kernel that you sent above?
Thank you!