相當於`tensorflow_probability.edward2`中的`ed.copy`

Question

我正在嘗試將我的代碼從edward更改為tensorflow_probability.edward2 。 問題是，每當我定義后驗分布時，我都會使用a_post = ed.copy(a, {u: qu}, scope='a_post')但.copy API 似乎不再可用：

module 'tensorflow_probability.python.edward2' has no attribute 'copy'

執行相同操作的tensorflow_probability方式是什么？

Answer 1

在 edward 中，復制依賴於不受支持的 TF 圖遍歷和復制。 在 edward2 中，該方法基於跟蹤，使用“攔截器”。 查看https://github.com/tensorflow/probability/blob/master/tensorflow_probability/examples/deep_exponential_family.py以獲取使用“磁帶”攔截器的 VI 示例。

更新：這可能是一個更簡單和/或更熟悉（LDA）的例子： https : //github.com/tensorflow/probability/blob/master/tensorflow_probability/examples/latent_dirichlet_allocation_edward2.py