antsnormflows.flows.neural_spline.coupling

Implementations of various coupling layers. Code taken from https://github.com/bayesiains/nsf

Classes

Coupling(mask, transform_net_create_fn[, ...])	A base class for coupling layers.
PiecewiseCoupling(mask, transform_net_create_fn)
PiecewiseRationalQuadraticCDF(shape[, ...])
PiecewiseRationalQuadraticCoupling(mask, ...)