bee606d2a668d462c9eea4cc23d7cfd8fbeb36a3,geomstats/geometry/hyperbolic_space.py,HyperbolicMetric,log,#HyperbolicMetric#Any#Any#,315

Before Change

                   - gs.einsum("ni,nj->nj", coef_2, base_point))
            return log

        if(self.point_type=="Poincare"):
            raise NotImplementedError(
                "The computation of the poincare log"
                " is not implemented.")


    def dist(self, point_a, point_b):
        
        Geodesic distance between two points.

After Change

        if self.point_type=="poincare":
            kpx = self.add(-base_point, point)

            norm_kpx = kpx.norm(2, -1, keepdim=True).expand_as(kpx)

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

            res = (1 - norm_base_point ** 2) * ((gs.arc_tanh(norm_kpx))) * (kpx / norm_kpx)

            if (0 != len((norm_kpx == 0).nonzero())):
                res[norm_kpx == 0] = 0

Instances