wasserstein_distance.torch

Torch implementation for Wasserstein distance scores.

Functions

compute_wasserstein_distance_score_torch

Computes the Wasserstein distance score using Torch Tensor.