https://medium.com/@monocosmo77/updates-on-wasserstein-barycenter-part10-machine-learning-28d5fab47efd