Use of torch.nn.functional.grid_sample()?

yes it does @Zichun_Zhang