https://medium.com/@monocosmo77/working-with-wasserstein-distances-in-deep-learning-part2-machine-learning-186521dbfc3a