So it sounds like the final step in your model is a custom probability distribution, rather than one of the built in tf ones that edward alias into
In that case, you’ll need to subclass
tf.Distribution and implement both
_log_prob, then use that to create a
RandomVariable class. Even though you’re not sampling from it, Edward needs
sample to create a tensor representation of the distribution, which it uses for inference.
Hard to be more specific without a code sample though.