sakura.utils.sliced_wasserstein

Sliced Wasserstein distance for loss computation

Reference:

Adapted from https://github.com/skolouri/swae

Classes

SlicedWasserstein

Computes Sliced Wasserstein Distance between encoded samples and target distribution