Эквивалент `ed.copy` в` tenorflow_probability.edward2`

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

В Эдварде копирование зависело от неподдерживаемого обхода и копирования TF-графа.В edward2 этот подход основан на отслеживании с использованием «перехватчиков».Посмотрите https://github.com/tensorflow/probability/blob/master/tensorflow_probability/examples/deep_exponential_family.py пример ВП с использованием перехватчика 'tape'.

Обновление: это может быть более простой и / или более знакомый (LDA) пример: https://github.com/tensorflow/probability/blob/master/tensorflow_probability/examples/latent_dirichlet_allocation_edward2.py