3763f47d3ef6543c0b12ab0e08213baad5f63153,geomstats/geometry/hyperbolic_space.py,HyperbolicMetric,exp,#HyperbolicMetric#Any#Any#,521

Before Change

        elif self.point_type == "ball":

            norm_base_point = base_point.norm(2,
                                              -1, keepdim=True).expand_as(
                                                base_point)

After Change

        elif self.point_type == "ball":

            norm_base_point = gs.to_ndarray(gs.norm(base_point, -1), 2, -1)
            norm_base_point = gs.repeat(norm_base_point,
                                        base_point.shape[-1], -1)
            den = 1 - norm_base_point**2

            norm_tan = gs.to_ndarray(gs.norm(tangent_vec, -1), 2, -1)
            norm_tan = gs.repeat(norm_tan, base_point.shape[-1], -1)

            lambda_base_point = 1 / den

            direction = tangent_vec / norm_tan

            factor = gs.tanh(lambda_base_point * norm_tan)

Instances