072e5680310c1cefef7e520c16bb20e92da3016d,geomstats/geometry/hypersphere.py,HypersphereMetric,exp,#HypersphereMetric#Any#Any#,342

Before Change

            tangent_vec, base_point)
        norm_tangent_vec = self.embedding_metric.norm(proj_tangent_vec)

        n_tiles_vec = gs.cast(gs.divide(n_exps, n_tangent_vecs), gs.int32)
        norm_tangent_vec = gs.tile(norm_tangent_vec, [n_tiles_vec, 1])
        n_tiles_base_point = gs.cast(
            gs.divide(n_exps, n_base_points), gs.int32)
        base_point = gs.tile(base_point, [n_tiles_base_point, 1])

After Change

                coef_1 = gs.squeeze(coef_1, axis=0)
                einsum_str = "i,nj->nj"
            elif n_base_points == 1:
                base_point = gs.squeeze(base_point, axis=0)
                einsum_str = "ni,j->nj"
            else:
                raise ValueError("Shape mismatch in einsum.")

Instances