https://medium.com/@monocosmo77/updates-on-wasserstein-barycenter-part9-machine-learning-ed4881976ef4