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