zuko.flows.core¶

Core building blocks.

Warning

This sub-module is deprecated and will be removed in the future. Use zuko.lazy instead.

Classes¶

Creates a lazy normalizing flow.

Descriptions¶

class zuko.flows.core.Flow(transform, base)¶

Creates a lazy normalizing flow.

See also

zuko.distributions.NormalizingFlow

Parameters:

transform (Union[LazyTransform, Sequence[LazyTransform]]) – A lazy transformation or sequence of lazy transformations.
base (LazyDistribution) – A lazy distribution.

forward(c=None)¶

Parameters:: c (Tensor | None) – A context \(c\).
Returns:: A normalizing flow \(p(X | c)\).
Return type:: NormalizingFlow