https://medium.com/@monocosmo77/updates-on-wasserstein-barycenter-part6-machine-learning-47aa68c170b8