Implement Wasserstein Critic Loss | TensorTonic