cdc013766a961e645e6e0e5284d18f9f2153b933,geomstats/hyperbolic_space.py,HyperbolicMetric,exp,#HyperbolicMetric#Any#Any#,179

Before Change

        if not gs.allclose(diff, 0, atol=TOLERANCE):
            tangent_vec = projected_tangent_vec
            logging.warning(
                "The input vector is not tangent to the hyperbolic space."
                " We project it on the tangent space at base_point={}.".format(
                    base_point))

        sq_norm_tangent_vec = self.embedding_metric.squared_norm(
                tangent_vec)

After Change

        coef_1 = gs.squeeze(coef_1, axis=-1)
        coef_2 = gs.squeeze(coef_2, axis=-1)
        exp = (gs.einsum("...,...j->...j", coef_1, base_point)
               + gs.einsum("...,...j->...j", coef_2, tangent_vec))

        hyperbolic_space = HyperbolicSpace(dimension=self.dimension)
        exp = hyperbolic_space.regularize(exp)
        return exp

Instances