tfep.nn.transformers.affine

Affine transformer for autoregressive normalizing flows.

Functions

affine_transformer(x, shift, log_scale)

Implement an affine transformer for autoregressive normalizing flows.

affine_transformer_inverse(y, shift, log_scale)

Inverse function of affine_transformer.

volume_preserving_shift_transformer(x, shift)

Implement a volume-preserving transformer for autoregressive normalizing flows.

volume_preserving_shift_transformer_inverse(y, ...)

Inverse function of volume_preserving_shift_transformer.

Classes

AffineTransformer(*args, **kwargs)

Affine transformer module for autoregressive normalizing flows.

VolumePreservingShiftTransformer([...])

Implement a volume-preserving transformer for autoregressive normalizing flows.