flow_matching.utils.manifolds#

Manifold#

Manifold classes for logarithmic and exponential map projections

Manifold

A manifold class that contains projection operations and logarithm and exponential maps.

Sphere

Represents a hyperpshere in \(R^D\).

FlatTorus

Represents a flat torus on the \([0, 2\pi]^D\) subspace.

Utility Functions#

geodesic

Generate parameterized function for geodesic curve.