I find it is apparently different between the two APIs( tf.image.crop_and_resize vs torchvision.ops.roi_align). I provide the example here:
import numpy as np