permutations#

Permutations applied between the coupling layers of a CouplingFlow.

Classes

FixedPermutation(*args, **kwargs)

Interface class for permutations with no learnable parameters.

OrthogonalPermutation(*args, **kwargs)

Implements a learnable orthogonal transformation according to [1].

RandomPermutation(*args, **kwargs)

Random permutation of the target dimensions.

Swap(*args, **kwargs)

Cyclic shift of the target dimensions by half their number.