be291b6a6ab8a663beac72fefa213bc7216a1617,geomstats/learning/frechet_mean.py,,_ball_gradient_descent,#Any#Any#Any#Any#Any#Any#,177

Before Change

        while convergence > tau and max_iter > iteration:

            iteration += 1
            grad_tangent = 2 * log(torch.from_numpy(gs.repeat(barycenter, len(points), axis = 0)),
                                   torch.from_numpy(points)) * torch.from_numpy(weights)
            grad_tangent = grad_tangent.data.numpy()
            cc_barycenter = exp(torch.from_numpy(barycenter),

After Change

            barycenter_gs  = gs.squeeze(barycenter)
            points_gs = gs.squeeze(points)

            grad_tangent = gs.zeros((len(points),len(barycenter_gs),len(points[0][0])))


            for j in range(len(points)):
                for i in range(len(barycenter_gs)):

Instances