ott.neural.methods.flows.dynamics.BrownianBridge.compute_xt

BrownianBridge.compute_xt(rng, t, src, tgt)

Sample from the probability path.

Sample from the probability path between \(x_0\) and \(x_1\) at time \(t\).

Parameters:

• rng (Array) – Random number generator.

• 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:

Samples from the probability path between \(x_0\) and \(x_1\) at time \(t\).