I have images which have empty values, I want to fill the empty values with nearest neighbor.  What is an efficient method for this sort of imputation?
The images will already be pytorch tensors on GPU, and sending to cpu and back to gpu is not an option, so I cannot use numpy nor scikit options.