Extension to `F.grid_sample` that allows using batch index per grid point. - View it on GitHub