506b234094009d55d326f1ce957f53f15d4b65ee,geomstats/spd_matrices_space.py,SPDMetric,exp,#SPDMetric#Any#Any#,160

Before Change

        sqrt_base_point = scipy.linalg.sqrtm(base_point)
        inv_sqrt_base_point = np.linalg.inv(sqrt_base_point)

        tangent_vec_at_id = np.dot(np.dot(inv_sqrt_base_point,
                                          tangent_vec),
                                   inv_sqrt_base_point)
        exp_from_id = group_exp(tangent_vec_at_id)

        exp = np.dot(sqrt_base_point,
                     np.dot(exp_from_id,

After Change

                or n_tangent_vecs == 1
                or n_base_points == 1)

        sqrt_base_point = np.zeros((n_base_points, mat_dim, mat_dim))
        for i in range(n_base_points):
            sqrt_base_point[i] = scipy.linalg.sqrtm(base_point[i])

        inv_sqrt_base_point = np.linalg.inv(sqrt_base_point)

        tangent_vec_at_id = np.matmul(inv_sqrt_base_point,
                                      tangent_vec)

Instances