antsnormflows.distributions.decoder

Classes

BaseDecoder()

NNBernoulliDecoder(net)

BaseDecoder representing a Bernoulli distribution with mean parametrized by a NN

NNDiagGaussianDecoder(net)

BaseDecoder representing a diagonal Gaussian distribution with mean and std parametrized by a NN