antsnormflows.flows.reshape

Classes

Merge([mode])

Same as Split but with forward and backward pass interchanged

Split([mode])

Split features into two sets

Squeeze2d()

Squeeze operation of multi-scale architecture (RealNVP / Glow), which trades spatial resolution for more channels.

Squeeze3d()

Squeeze operation for 3D images, extending the concept from Glow.