Using affine_grid to implement the ROIAlign failed

Maybe you want to take a look at the PyTorch implementation.
Also, this repository implemented the RoI Align using affine_grid and grid_sample.

I got these links from this discussion.