hypll.tensors.manifold_parameter

Classes

ManifoldParameter

class hypll.tensors.manifold_parameter.ManifoldParameter
static __new__(cls, data, manifold, man_dim, requires_grad=True)
__init__(data, manifold: Manifold, man_dim: int = -1, requires_grad: bool = True) None

Creates an instance of ManifoldTensor.

Parameters:
  • data – Torch tensor of points on the manifold.

  • manifold – Manifold instance.

  • man_dim – Dimension along which points are on the manifold. -1 by default.

TODO(Philipp, 05/23): Let’s get rid of requires_grad if possible.