https://medium.com/@monocosmo77/how-unbalanced-optimal-transport-works-part10-machine-learning-2024-59eed2e386da