我正在尝试复制我在GoogleCol堕胎(这里)上找到的代码片段,并且我在文档中找不到所使用的方法(tfb.Shift和tfp.util.TransformedVariable)有问题。
import numpy as np
import tensorflow.compat.v2 as tf
import tensorflow_probability as tfp
tfb = tfp.bijectors
tfd = tfp.distributions
tf.enable_v2_behavior()
constrain_positive = tfb.Shift(np.finfo(np.float64).tiny)(tfb.Exp())
amplitude_var = tfp.util.TransformedVariable(
initial_value=1.,
bijector=constrain_positive,
name='amplitude',
dtype=np.float64)我得到了
AttributeError: module 'tensorflow_probability.python.bijectors' has no attribute 'Shift'和
AttributeError: module 'tensorflow_probability.python.util' has no attribute 'TransformedVariable'可能是版本问题?
我使用的是tensorflow 2.0.0和tensorflow-probability 0.8.0,我看到colab笔记本使用的是@tf.function,所以我认为它也使用了tensorflow 2.0.0。
我能用什么代替?
非常感谢
发布于 2020-01-04 16:13:04
不幸的是,在tensorflow_probability的依赖关系管理中似乎存在很多问题。我尝试了您的代码,得到了一个稍微不同的导入错误。
您可以尝试以下解决方法:
pip install tensorflow==2.1.0rc2https://stackoverflow.com/questions/59578283
复制相似问题