https://medium.com/@monocosmo77/updates-on-wasserstein-barycenter-part8-machine-learning-d59ed178d7d7