hypll.nn.modules.change_manifold

Classes

ChangeManifold

Changes the manifold of the input manifold tensor to the target manifold.

class hypll.nn.modules.change_manifold.ChangeManifold

Changes the manifold of the input manifold tensor to the target manifold.

target_manifold

Manifold the output tensor should be on.

__init__(target_manifold: Manifold)

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

forward(x: ManifoldTensor) → ManifoldTensor

Changes the manifold of the input tensor to self.target_manifold.

By default, applies logmap and expmap at the origin to switch between manifold. Applies shortcuts if possible.

Parameters:

x – Input manifold tensor.

Returns:

Tensor on the target manifold.