flow_matching.path#

Probability Paths#

ProbPath

Abstract class, representing a probability path.

AffineProbPath

The AffineProbPath class represents a specific type of probability path where the transformation between distributions is affine.

CondOTProbPath

The CondOTProbPath class represents a conditional optimal transport probability path.

MixtureDiscreteProbPath

The MixtureDiscreteProbPath class defines a factorized discrete probability path.

GeodesicProbPath

The GeodesicProbPath class represents a specific type of probability path where the transformation between distributions is defined through the geodesic path.

Path Sample#

Corresponds to an instance of a sample drawn from the probability path.

path_sample.PathSample

Represents a sample of a conditional-flow generated probability path.

path_sample.DiscretePathSample

Represents a sample of a conditional-flow generated discrete probability path.