ott.geometry.costs.Bures.scale_covariances#

Bures.scale_covariances(cov_sqrt, cov_i, lambda_i)[source]#

Vectorized version of scale_covariances. Takes similar arguments as scale_covariances but with additional array axes over which scale_covariances is mapped.

Original documentation:

Iterate update needed to compute barycenter of covariances.