gensbi.diffusion.path.path_sample#

Classes#

EDMPathSample

Represents a sample of a diffusion generated probability path.

Module Contents#

class gensbi.diffusion.path.path_sample.EDMPathSample[source]#

Represents a sample of a diffusion generated probability path.

x_1[source]#

the target sample \(X_1\).

Type:: Array

sigma[source]#

the noise scale \(t\).

Type:: Array

x_t[source]#

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

Type:: Array

get_batch()[source]#

Returns the batch as a tuple (x_1, x_t, sigma).

Returns:: The target sample, the noisy sample, and the noise scale.
Return type:: Tuple[Array, Array, Array]

sigma: jax.Array[source]#

x_1: jax.Array[source]#

x_t: jax.Array[source]#