从 tensorflow 1.x 升级到 2.0

Question

我是张量流的新手。 试过这个简单的例子：

import tensorflow as tf
sess = tf.Session()
x = tf.placeholder(tf.float32)
y = tf.placeholder(tf.float32)
z = x + y
print(sess.run(z, feed_dict={x: 3.0, y: 4.5}))

得到了一些警告 The name tf.Session is deprecated. Please use tf.compat.v1.Session instead. 和正确答案 - 7.5

在阅读here 后，我了解到警告是由于从 tf 1.x 升级到 2.0 造成的，描述的步骤“简单”但没有给出任何示例......

我试过了：

@tf.function
def f1(x1, y1):
    return tf.math.add(x1, y1)


print(f1(tf.constant(3.0), tf.constant(4.5)))

1. 我的代码是否正确（在链接中定义的意义上）？
2. 现在，我得到了Tensor("PartitionedCall:0", shape=(), dtype=float32) 作为输出，我怎样才能得到实际值？

Answer 1

您的代码确实是正确的。您收到的警告表明，从 Tensorflow 2.0 开始，API 中将不存在tf.Session()。因此，如果您希望您的代码与 Tensorflow 2.0 兼容，您应该改用tf.compat.v1.Session。因此，只需更改此行：

sess = tf.Session()

收件人：

sess = tf.compat.v1.Session()

然后，即使您将 Tensorflow 从 1.xx 更新到 2.xx，您的代码也会以相同的方式执行。至于Tensorflow 2.0中的代码：

@tf.function
def f1(x1, y1):
    return tf.math.add(x1, y1)

print(f1(tf.constant(3.0), tf.constant(4.5)))

如果你在 Tensorflow 2.0 中运行它就可以了。如果你想在不安装 TensorFlow 2.0 的情况下运行相同的代码，可以执行以下操作：

import tensorflow as tf
tf.enable_eager_execution()

@tf.function
def f1(x1, y1):
    return tf.math.add(x1, y1)

print(f1(tf.constant(3.0), tf.constant(4.5)).numpy())

这样做的原因是因为从 Tensorflow 2.0 开始执行 Tensorflow 操作的默认方式是在 Eager 模式下。在 Tensorflow 1.xx 中激活 Eager 模式的方法是在导入 Tensorflow 之后立即启用它，就像我在上面的示例中所做的那样。

Answer 2

根据 Tensorflow 2.0，您的代码是正确的。 Tensorflow 2.0 与 numpy 的结合更加紧密，所以如果你想得到操作的结果，你可以使用numpy() 方法：

print(f1(tf.constant(3.0), tf.constant(4.5)).numpy())