https://medium.com/@monocosmo77/working-with-wasserstein-2-metric-in-machine-learning-research-part4-4c4ad695dfbe