TensorFlow Sinkhorn Algorithm

Optimal transport of particles in real-time with the TensorFlow C++ API.

Although this example transports particles in 2D space, I’ve also added a feature for 3D space.

The Algorithm

Alex Mordvintsev, who’s widely known for developing DeepDream shared this animation and Python notebook. His code is written with the TensorFlow Python API, which I converted to the TensorFlow C++ API.