Can torchvision.ops.RoIAlign be used instead of RoIAlign（CUDA extensions）

[alexandsunny]

I had some problems converting my model to onnx Can torchvision.ops.RoIAlign be used instead of RoIAlign（CUDA extensions）

[xlliu7]

I guess it is okay after some modification. Note that torchvision.ops.RoIAlign only works for 2D images (4D feature maps of shape NxCxHxW). So you should resize the input feature sequence of the RoIAlign module from NxCxT to NxCx1xT and also modify the RoI locations to fit the new shape.