Closest pair of points problem optimized with tensorflow?

I am a novice when it comes to tensorflow and AI.

I wrote a solution to the problem first with numpy, then numba optimized. When it comes to tensors, I have no idea how to even start the optimization. Are there some easy to follow tutorials out there? How do I even tell tensorflow the rules and goal of the optimization?

https://github.com/XeonPhobia/Pixel-replacement-algorithm-based-on-RMSE/blob/main/test.py