hypll.manifolds.poincare_ball.math.linalg

Functions

poincare_fully_connected(x, z, bias, c[, dim])	The Poincare fully connected layer operation.
poincare_hyperplane_dists(x, z, r, c[, dim])	The Poincare signed distance to hyperplanes operation.

hypll.manifolds.poincare_ball.math.linalg.poincare_hyperplane_dists(x: Tensor, z: Tensor, r: Tensor | None, c: Tensor, dim: int = -1) → Tensor

The Poincare signed distance to hyperplanes operation.

Parameters

xtensor

contains input values

ztensor

contains the hyperbolic vectors describing the hyperplane orientations

rtensor

contains the hyperplane offsets

ctensor

curvature of the Poincare disk

Returns

tensor

signed distances of input w.r.t. the hyperplanes, denoted by v_k(x) in the HNN++ paper

hypll.manifolds.poincare_ball.math.linalg.poincare_fully_connected(x: Tensor, z: Tensor, bias: Tensor | None, c: Tensor, dim: int = -1) → Tensor

The Poincare fully connected layer operation.

Parameters

xtensor

contains the layer inputs

ztensor

contains the hyperbolic vectors describing the hyperplane orientations

biastensor

contains the biases (hyperplane offsets)

ctensor

curvature of the Poincare disk

Returns

tensor

Poincare FC transformed hyperbolic tensor, commonly denoted by y