ott.neural.methods.flows.dynamics.ConstantNoiseFlow.compute_ut

ConstantNoiseFlow.compute_ut(t, src, tgt)

Evaluate the conditional vector field.

Evaluate the conditional vector field defined between \(x_0\) and \(x_1\) at time \(t\).

Parameters:

• t (Array) – Time \(t\) of shape [batch, 1].

• src (Array) – Sample from the source distribution of shape [batch, ...].

• tgt (Array) – Sample from the target distribution of shape [batch, ...].

Return type:

Array

Returns:

Conditional vector field evaluated at time \(t\).