- class ott.geometry.costs.Bures(dimension, **kwargs)#
Bures distance between a pair of (mean, cov matrix) raveled as vectors.
Compute matrix of all costs (including norms) for vectors in x / y.
Compute matrix of all pairwise-costs (no norms) for vectors in x / y.
Compute the Bures barycenter of weighted Gaussian distributions.
covariance_fixpoint_iter(covs, lambdas[, rtol])
Iterate fix-point updates to compute barycenter of Gaussians.
Vectorized version of means_and_covs_to_x.
Compute norm of Gaussian, sq.
Compute - 2 x Bures dot-product.
Monitor change in two successive estimates of matrices.
scale_covariances(cov_sqrt, cov_i, lambda_i)
Vectorized version of scale_covariances.
Extract mean and covariance matrix from raveled d(1 + d) vector.