hypll.manifolds.base.manifold

Classes

Manifold

class hypll.manifolds.base.manifold.Manifold

__init__() → None

Initialize internal Module state, shared by both nn.Module and ScriptModule.

abstract project(x: ManifoldTensor, eps: float = -1.0) → ManifoldTensor

abstract expmap(v: TangentTensor) → ManifoldTensor

abstract logmap(x: ManifoldTensor | None, y: ManifoldTensor) → TangentTensor

abstract transp(v: TangentTensor, y: ManifoldTensor) → TangentTensor

abstract dist(x: ManifoldTensor, y: ManifoldTensor) → Tensor

abstract cdist(x: ManifoldTensor, y: ManifoldTensor) → Tensor

abstract euc_to_tangent(x: ManifoldTensor, u: ManifoldTensor) → TangentTensor

abstract hyperplane_dists(x: ManifoldTensor, z: ManifoldTensor, r: Tensor | None) → Tensor

abstract fully_connected(x: ManifoldTensor, z: ManifoldTensor, bias: Tensor | None) → ManifoldTensor

abstract frechet_mean(x: ManifoldTensor, batch_dim: int | list[int] = 0, keepdim: bool = False) → ManifoldTensor

abstract midpoint(x: ManifoldTensor, batch_dim: int | list[int] = 0, w: Tensor | None = None, keepdim: bool = False) → ManifoldTensor

abstract frechet_variance(x: ManifoldTensor, mu: ManifoldTensor | None = None, batch_dim: int | list[int] = -1, keepdim: bool = False) → Tensor

abstract construct_dl_parameters(in_features: int, out_features: int, bias: bool = True) → ManifoldParameter | tuple[ManifoldParameter, Parameter]

abstract reset_parameters(weight: ManifoldParameter, bias: Parameter) → None

abstract flatten(x: ManifoldTensor, start_dim: int = 1, end_dim: int = -1) → ManifoldTensor

abstract unfold(input: ManifoldTensor, kernel_size: _size_2_t, dilation: _size_2_t = 1, padding: _size_2_t = 0, stride: _size_2_t = 1) → ManifoldTensor

abstract cat(manifold_tensors: Tuple[ManifoldTensor, ...] | List[ManifoldTensor], dim: int = 0) → ManifoldTensor