hypll.nn.modules.flatten

Classes

HFlatten

Flattens a contiguous range of dims into a tensor.

class hypll.nn.modules.flatten.HFlatten

Flattens a contiguous range of dims into a tensor.

start_dim

First dimension to flatten (default = 1).

end_dim

Last dimension to flatten (default = -1).

__init__(start_dim: int = 1, end_dim: int = -1)

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

forward(x: ManifoldTensor) → ManifoldTensor

Flattens the manifold input tensor.