More DistributionLambda in layers
Closed this issue · 1 comments
vdutor commented
LikelihoodLayer
could return the correctly parametrized y-distribution (perhaps a mixture if using multiple samples?)LatentVariableLayer
could be atfp.DistributionLambda
and return prior/posterior as appropriate (we'd then need to take out again the composition with inputs through addition or concatenation).Encoder
could also return the distribution directly instead of keeping track of the class of the prior.