tfep.nn.transformers.spline

Circular spline transformer for autoregressive normalizing flows.

Functions

neural_spline_transformer(x, x0, y0, widths, ...)	Implement the neural spline transformer.
neural_spline_transformer_inverse(y, x0, y0, ...)	Implement the inverse of the neural spline transformer.

Classes

NeuralSplineTransformer(x0, xf, n_bins[, ...])

Neural spline transformer module for autoregressive normalizing flows.