antsnormflows.flows.residual

Functions

basic_logdet_estimator(g, x, n_power_series, ...)
batch_jacobian(g, x)
batch_trace(M)
geometric_1mcdf(p, k, offset)
geometric_sample(p, n_samples)
mem_eff_wrapper(estimator_fn, gnet, x, ...)
neumann_logdet_estimator(g, x, ...)
poisson_1mcdf(lamb, k, offset)
poisson_sample(lamb, n_samples)
safe_detach(tensor)

Classes

MemoryEfficientLogDetEstimator(*args, **kwargs)
Residual(net[, reverse, reduce_memory, ...])	Invertible residual net block (Optimized with deterministic power series truncation)
iResBlock(nnet[, n_power_series, ...])