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.