ott.neural.methods.flows.dynamics.StraightFlow

class ott.neural.methods.flows.dynamics.StraightFlow(sigma)

Base class for flows with straight paths.

Parameters:

sigma (float) – Noise used for computing time-dependent noise schedule.

Methods

compute_mu_t(t, src, tgt)	Compute the mean of the probability path.
compute_sigma_t(t)	Compute the standard deviation of the probability path at time \(t\).
compute_ut(t, src, tgt)	Evaluate the conditional vector field.
compute_xt(rng, t, src, tgt)	Sample from the probability path.