https://medium.com/@monocosmo77/updates-on-wasserstein-barycenter-part7-machine-learning-5dbb24efea15