eb52df01d6fa9beb97d367ff091d32186c91ea95,geomstats/geometry/poincare_ball.py,PoincareBallMetric,exp,#PoincareBallMetric#Any#Any#,87

Before Change

        tangent_vec = gs.to_ndarray(tangent_vec, to_ndim=2)
        base_point = gs.to_ndarray(base_point, to_ndim=2)

        if base_point.shape[0] == 1:
            base_point = gs.repeat(base_point, tangent_vec.shape[0], axis=0)

        if tangent_vec.shape[0] == 1:
            tangent_vec = gs.repeat(tangent_vec, base_point.shape[0], axis=0)

        norm_base_point = gs.to_ndarray(
            gs.linalg.norm(base_point, axis=-1), 2, axis=-1)
        norm_base_point = gs.to_ndarray(norm_base_point, to_ndim=2)

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

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

        lambda_base_point = 1 / den

        zero_tan = gs.isclose((tangent_vec * tangent_vec).sum(axis=-1), 0.)

        if norm_tan[zero_tan].shape[0] != 0:
            norm_tan[zero_tan] = EPSILON

        direction = tangent_vec / norm_tan

        factor = gs.tanh(lambda_base_point * norm_tan)

        exp = self.mobius_add(base_point, direction * factor)

After Change

        zero_tan = gs.isclose((tangent_vec * tangent_vec).sum(axis=-1), 0.)

        if gs.any(zero_tan):
            if norm_tan[zero_tan].shape[0] != 0:
                norm_tan[zero_tan] = EPSILON

        direction = gs.einsum("...i,...k->...i", tangent_vec, 1 / norm_tan)

        factor = gs.tanh(lambda_base_point * norm_tan)

Instances