antsnormflows.flows.affine.coupling

Classes

AffineConstFlow(shape[, scale, shift])

scales and shifts with learned constants per dimension.

AffineCoupling(param_map[, scale, ...])

Affine Coupling layer as introduced RealNVP paper, see arXiv: 1605.08803

AffineCouplingBlock(param_map[, scale, ...])

Affine Coupling layer including split and merge operation

CCAffineConst(shape, num_classes)

Affine constant flow layer with class-conditional parameters

MaskedAffineFlow(b[, t, s, s_cap, ...])

RealNVP as introduced in [arXiv: 1605.08803](https://arxiv.org/abs/1605.08803)