tensorflow-probability: AttributeError: Tensor.op is meaningless when eager execution is enabled

Question

I want to build a classification model that returns a distribution over probabilities for each class. I.e., instead of getting a single probability that a class is positive,