315990901cdcdca241013ee5e6c581391bae2523,geomstats/geometry/spd_matrices.py,SPDMetricBuresWasserstein,inner_product,#SPDMetricBuresWasserstein#,697

Before Change

        rotated_tangent_vec_a = gs.matmul(transp_eigvecs, tangent_vec_a)
        rotated_tangent_vec_a = gs.matmul(rotated_tangent_vec_a, eigvecs)
        rotated_tangent_vec_b = gs.matmul(transp_eigvecs, tangent_vec_b)
        rotated_tangent_vec_b = gs.matmul(rotated_tangent_vec_b, eigvecs)

        ones = gs.ones(eigvals.shape)
        vertical_index = gs.einsum("...i,...j->...ij", eigvals, ones)
        horizontal_index = gs.einsum("...j,...i->...ij", eigvals, ones)

After Change

            Inner-product.
        
        eigvals, eigvecs = gs.linalg.eigh(base_point)
        transp_eigvecs = Matrices.transpose(eigvecs)
        rotated_tangent_vec_a = Matrices.mul(transp_eigvecs, tangent_vec_a,
                                             eigvecs)
        rotated_tangent_vec_b = Matrices.mul(transp_eigvecs, tangent_vec_b,
                                             eigvecs)

Instances