3da4b071f1257d7d13e5f2100ed04cc30c37dc23,geomstats/hyperbolic_space.py,HyperbolicSpace,belongs,#HyperbolicSpace#Any#Any#,57

Before Change

            return False

        sq_norm = self.embedding_metric.squared_norm(point)
        euclidean_sq_norm = gs.dot(point, point.transpose())
        diff = gs.abs(sq_norm + 1)
        return diff < tolerance * euclidean_sq_norm

After Change

        sq_norm = self.embedding_metric.squared_norm(point)
        euclidean_sq_norm = gs.einsum("ij,ij->i", point, point)
        euclidean_sq_norm = gs.to_ndarray(euclidean_sq_norm, to_ndim=2, axis=1)
        diff = gs.abs(sq_norm + 1)

        return diff < tolerance * euclidean_sq_norm

Instances