ott.geometry.costs.Bures.covariance_fixpoint_iter#

Bures.covariance_fixpoint_iter(covs, lambdas, rtol=0.01)[source]#

Iterate fix-point updates to compute barycenter of Gaussians.

Return type

Array

Parameters
  • covs (jax._src.basearray.Array) –

  • lambdas (jax._src.basearray.Array) –

  • rtol (float) –