Can we expect a Nearest Neighbour Volumetric Grid Sampler in pytorch anytime soon?
Yes: in review at: https://github.com/pytorch/pytorch/pull/10051
Also, note that NN interpolation doesn’t have grad on grid by definition. 
1 Like