I am currently working on a VAE using keras and tensorflow/tensorflow-probability. I am using mnist as a training set. My problem here is the sampling of the input from p(x|
