python-3.x - 如何在 tensorflow 版本 2 中转换以下内容

Question

%tensorflow_version 1.x
import tensorflow as tf
print(tensorflow.__version__)
w = tf.Variable(0, dtype = tf.float32)
cost = w**2 - 8*w + 16
train = tf.train.GradientDescentOptimizer(learning_rate=0.01).minimize(cost)
init = tf.global_variables_initializer()
session = tf.Session()
session.run(init)
for i in range(1000):
    session.run(train)

print(session.run(w))
#3.994

我用谷歌搜索了这些东西，但没有得到任何结论，当我使用 tf-vr-2 时，该程序会计算多项式的因子

标签： python-3.xtensorflow

Answer 1

import tensorflow as tf

w = tf.Variable(0, dtype = tf.float32)

@tf.function
def cost():
    return w**2 - 8*w + 16


train = tf.compat.v1.train.GradientDescentOptimizer(learning_rate=0.01)
for i in range(1000):
    train.minimize(loss=cost, var_list=[w])

tf.print(w)