https://medium.com/@monocosmo77/working-with-sliced-wasserstein-concept-part2-machine-learning-2024-955d31cb7722