https://medium.com/@monocosmo77/working-with-wasserstein-distance-in-machine-learning-problems-part3-a64e749c4127