tensorflow / probability

Probabilistic reasoning and statistical analysis in TensorFlow

Home Page:https://www.tensorflow.org/probability/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Distribution Output Cannot Be Passed to Keras Layers

i418c opened this issue · comments

In the nightly versions of TF and TFP, it's no longer possible to pass the output of a distribution into the call function of a Keras layer as a positional argument.
See the gist reproducing this here.

This is also the result of TFP not being compatible with Keras 3. Further info can be found in #1780.