antsnormflows.flows.mixing

Classes

Invertible1x1Conv(num_channels[, use_lu, s_cap])	Invertible 1x1 convolution introduced in the Glow paper.
Invertible1x1x1Conv(num_channels[, use_lu, ...])	Invertible 1x1x1 convolution introduced in the Glow paper Assumes 5d input/output tensors of the form NCHWD
InvertibleAffine(num_channels[, use_lu])	Invertible affine transformation without shift, i.e. one-dimensional version of the invertible 1x1 convolutions.
LULinearPermute(num_channels[, identity_init])	Fixed permutation combined with a linear transformation parametrized using the LU decomposition, used in https://arxiv.org/abs/1906.04032
Permute(num_channels[, mode])	Permutation features along the channel dimension