315990901cdcdca241013ee5e6c581391bae2523,geomstats/geometry/spd_matrices.py,SPDMetricBuresWasserstein,exp,#SPDMetricBuresWasserstein#,734

Before Change

            Riemannian exponential.
        
        eigvals, eigvecs = gs.linalg.eigh(base_point)
        transp_eigvecs = gs.einsum("...ij->...ji", eigvecs)
        rotated_tangent_vec = gs.matmul(transp_eigvecs, tangent_vec)
        rotated_tangent_vec = gs.matmul(rotated_tangent_vec, eigvecs)

        ones = gs.ones(eigvals.shape)

After Change

            Riemannian exponential.
        
        eigvals, eigvecs = gs.linalg.eigh(base_point)
        transp_eigvecs = Matrices.transpose(eigvecs)
        rotated_tangent_vec = Matrices.mul(transp_eigvecs, tangent_vec,
                                           eigvecs)
        coefficients = 1 / (eigvals[..., :, None] + eigvals[..., None, :])
        rotated_sylvester = rotated_tangent_vec * coefficients

Instances