hypll.utils.layer_utils

Functions

check_if_man_dims_match(layer, man_dim, input)
check_if_manifolds_match(layer, input)
op_in_tangent_space(op, manifold, input)

hypll.utils.layer_utils.check_if_manifolds_match(layer: Module, input: ManifoldTensor) → None

hypll.utils.layer_utils.check_if_man_dims_match(layer: Module, man_dim: int, input: ManifoldTensor) → None

hypll.utils.layer_utils.op_in_tangent_space(op: Callable, manifold: Manifold, input: ManifoldTensor) → ManifoldTensor