506b234094009d55d326f1ce957f53f15d4b65ee,geomstats/spd_matrices_space.py,SPDMetric,log,#SPDMetric#Any#Any#,182

Before Change

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

        point_near_id = np.dot(np.dot(inv_sqrt_base_point,
                                      point),
                               inv_sqrt_base_point)
        log_at_id = group_log(point_near_id)

        log = np.dot(np.dot(sqrt_base_point,
                            log_at_id),

After Change

            point = np.expand_dims(point, axis=0)
        assert point.ndim == 3

        if base_point.ndim == 2:
            base_point = np.expand_dims(base_point, axis=0)
        assert base_point.ndim == 3

        n_points, _, _ = point.shape
        n_base_points, mat_dim, _ = base_point.shape

        assert (n_points == n_base_points
                or n_points == 1
                or n_base_points == 1)

Instances