Wasserstein Distance using neural network to calculate wasserstein distance between two data sets. Example 2 distributions above should return a w_distance of θ