The closest analog in PyTorch is torch.nn.functional.grid_sample() 58, but this is actually closer to the dense_image_warp function.
The conventions for the grid are also slightly different in PyTorch, but you can probably do a conversion to fit your needs.
I hope this helps!