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

Before Change

        // TODO(nina): Raise error when vector is not tangent
        n_base_points, extrinsic_dim = base_point.shape
        n_tangent_vecs, _ = tangent_vec.shape
        n_exps = gs.cast(gs.maximum(n_base_points, n_tangent_vecs), gs.int32)

        hypersphere = Hypersphere(dimension=extrinsic_dim-1)
        proj_tangent_vec = hypersphere.projection_to_tangent_space(
            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

        n_coef_1 = n_tangent_vecs
        if n_coef_1 != n_base_points:
            if n_coef_1 == 1:
                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.")

        exp = (gs.einsum(einsum_str, coef_1, base_point)
               + gs.einsum("ni,nj->nj", coef_2, proj_tangent_vec))

        return exp

Instances