e4833d449452dc9b47a4e23c22c4b2b8abc7a682,geomstats/learning/exponential_barycenter.py,,_default_gradient_descent,#Any#Any#Any#Any#Any#Any#Any#,14

Before Change

    while iteration < max_iter:
        if not (grad_norm > epsilon or iteration == 0):
            break
        centered_points = group.compose(group.inverse(mean), points)
        logs = group.log_from_identity(point=centered_points)
        tangent_mean = step * gs.einsum(
            "n, nk...->k...", weights / sum_weights, logs)

After Change

    while iteration < max_iter:
        if not (grad_norm > epsilon or iteration == 0):
            break
        inv_mean = group.inverse(mean)
        print(inv_mean.shape)
        print(points.shape)
        centered_points = group.compose(inv_mean, points)
        logs = group.log_from_identity(point=centered_points)
        tangent_mean = step * gs.einsum(
            "n, nk...->k...", weights / sum_weights, logs)
        mean_next = group.compose(

Instances