https://medium.com/@monocosmo77/updates-on-wasserstein-barycenter-part11-machine-learning-b35e2f0cd03c