https://medium.com/@monocosmo77/how-unbalanced-optimal-transport-works-part9-machine-learning-2024-aa9700fb1fc9