ott.neural.methods.flows.dynamics.StraightFlow.compute_sigma_t

ott.neural.methods.flows.dynamics.StraightFlow.compute_sigma_t#

abstract StraightFlow.compute_sigma_t(t)#

Compute the standard deviation of the probability path at time \(t\).

Parameters:

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

Return type:

Array

Returns:

Standard deviation of the probability path at time \(t\).