ott.neural.methods.flows.dynamics.BaseFlow.compute_xt# BaseFlow.compute_xt(rng, t, src, tgt)[source]# 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\).