gensbi.flow_matching.path.path_sample#

Classes#

PathSample

Represents a sample of a conditional-flow generated probability path.

Module Contents#

class gensbi.flow_matching.path.path_sample.PathSample[source]#

Represents a sample of a conditional-flow generated probability path.

x_1[source]#

the target sample \(X_1\).

Type:

Array

x_0[source]#

the source sample \(X_0\).

Type:

Array

t[source]#

the time sample \(t\).

Type:

Array

x_t[source]#

samples \(X_t \sim p_t(X_t)\), shape (batch_size, …).

Type:

Array

dx_t[source]#

conditional target \(\frac{\partial X}{\partial t}\), shape: (batch_size, …).

Type:

Array

dx_t: jax.Array[source]#
t: jax.Array[source]#
x_0: jax.Array[source]#
x_1: jax.Array[source]#
x_t: jax.Array[source]#