antsnormflows.flows.base

Functions

stable_fp_dtype(x)

Return a numerically stable floating-point dtype for exp/log-scale computations: at least float32, but preserves float64 when the input is already double precision.

zero_log_det_like_z(z)

Classes

Composite(flows)

Composes several flows into one, in the order they are given.

Flow()

Generic class for flow functions

Reverse(flow)

Switches the forward transform of a flow layer with its inverse and vice versa