ott.neural.methods.flows.dynamics.StraightFlow.compute_mu_t# StraightFlow.compute_mu_t(t, src, tgt)[source]# Compute the mean of the probability path. Compute the mean of the probability path 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