graph_diffusion_wasserstein