Advent of Code 2021 in pure TensorFlow - day 5

I’m going on with the challenge :slight_smile:

The day 5 challenge is easily solvable in pure TensorFlow thanks to its support for various distance functions and the power of the tf.math package. The problem only requires some basic math knowledge to be completely solved - and a little bit of computer vision experience doesn’t hurt.

Any feedback is welcome!