layers#

Coupling layers, which transform one half of the input conditioned on the other half.

Classes

DualCoupling(*args, **kwargs)

Dual coupling layer composed of two sequential single coupling layers.

SingleCoupling(*args, **kwargs)

Implements a single coupling layer as a composition of a subnet and a transform.