https://medium.com/@monocosmo77/how-unbalanced-optimal-transport-works-part1-machine-learning-2024-420401cf2633