Mapping values in a tensor

[Felix_Lessange (Damien MENIGAUX)]

I think this is what you’re looking for Cv2 remap in pytorch? - #10 by Felix_Lessange