TensorFlow梯度下降求解最小值

# tensorflow 求解f(x)=(x-1)^2+(y-2)^2+10的最小值

import tensorflow as tf

# 定义变量
w = tf.Variable(tf.constant([10.]));
b = tf.Variable(tf.constant([10.]));

sum = tf.add(tf.square(tf.add(w, tf.constant([-1.0]))), tf.square(tf.add(b, tf.constant([-2.0]))));
min = tf.add(sum, tf.constant([10.]));

optimizer = tf.train.GradientDescentOptimizer(0.5)

train = optimizer.minimize(min)

# 初始化变量
init = tf.initialize_all_variables()

# 启动图 (graph)
sess = tf.Session()
sess.run(init)

sess.run(train)
print(sess.run(w))
print(sess.run(b))
print(sess.run(min))
不如
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容