Hi I have a tensor of size (1,320,1216), which has values = 0 at most of the places.  I want to replace those values with their nearest neighbors. What is the most efficient and fast way of achieving this considering this would happen in online training?