How to use the grid_sample to warp the image