Wasserstein K -means for clustering probability distributions