Skip to contents

theta -> base variable u, accumulating the log |det Jacobian|. Returns list(u = (b, p) tensor, logdet = (b,) tensor).

Usage

maf_forward(net, theta, x)