ott.tools.gaussian_mixture.gaussian.Gaussian.w2_dist

ott.tools.gaussian_mixture.gaussian.Gaussian.w2_dist#

Gaussian.w2_dist(other)[source]#

Wasserstein distance \(W_2^2\) to another Gaussian.

\[W_2^2 = ||\mu_0-\mu_1||^2 + \text{trace} ( (\Lambda_0^\frac{1}{2} - \Lambda_1^\frac{1}{2})^2 )\]
Parameters:

other (Gaussian) – other Gaussian

Return type:

Array

Returns:

The \(W_2^2\) distance between self and other