ConstantinaNicolaou

Feature request: 2-D weights option for log prob

1

I am working on a VAE with a Gaussian decoder using tensorflow probability. I calculate the log likelihood using ```x_hat.log_prob(x)``` where ```x_hat``` is the output of the decoder (```tfp.distributions.Distribution```). However,...