ott.neural.methods.flows.dynamics.BaseFlow.compute_sigma_t

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

abstract BaseFlow.compute_sigma_t(t)[source]#

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\).