我目前正在上一门Coursera深度学习课程,并且给定的Tensorflow代码是用Tensorflow 1语法(uses会话)编写的,但我无法找到如何将其重写以与Tensorflow 2一起使用的方法。
这是非常简单的代码:
import numpy as np
import tensorflow as tf
y_hat = tf.constant(36, name='y_hat')
y = tf.constant(39, name='y')
loss = tf.Variable((y-y_hat)**2, name='loss')
init = tf.global_variables_initializer()
with tf.Session() as session:
session.run(init)
print(session.run(loss))
在哪里可以找到有关Tensorflow 1和2之间的MANUAL语法翻译的指南?谁能重写给定的代码以与Tensorflow 2一起使用(使用函数而不是sesion),因为我对Tensorflow还是很陌生的?先感谢您。
在tensorflow 2.0中,您可以将代码编写为
y_hat = tf.constant(36)
y = tf.constant(39)
loss = tf.Variable((y-y_hat)**2)
print(loss.numpy())
这是非常pythonic的,并且默认情况下启用了急切执行
假设您正在训练一个模型,然后有一个想要在每次迭代中调用的自定义函数,现在您可以通过将@tf.function()
函数放在前面来高效地执行此操作
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句