Manifold learning in Wasserstein space