使用Tensorflow求一个函数的最大值点

一、tensorflow 入门

import numpy as np
import tensorflow as tf

w = tf.Variable(0, dtype=tf.float32)
# cost = tf.add(tf.add(w**2, tf.multiply(-10., w)), 25)
cost = w**2 - 10 * w + 25 # TensorFlow会自动转换这些运算符！

train = tf.train.GradientDescentOptimizer(0.01).minimize(cost) # 学习率 和 成本函数

init = tf.global_variables_initializer() 
session = tf.Session()
session.run(init) # 只是执行了第二行代码！！
print(session.run(w))
0.0
session.run(train) # tensorflow 会根据 前向传播 自动生成 反向传播操作！
print(session.run(w))
0.099999994

for i in range(1000):
    session.run(train)
print(session.run(w))
4.9999886

二、placehold 占位符功能

x = tf.placeholder(tf.float32, [3, 1])

# cost = w**2 - 10 * w + 25
cost = x[0][0]*w**2 + x[1][0]*w + w[2][0]

coefficients = np.array([[1.], [-10.], [25.]])


session.run(train, feed_dict={x:coefficients})
for i in range(1000):
    session.run(train, feed_dict={x:coefficients})     # 每次训练自动填充掉x的占位符！
print(session.run(w))

coefficients = np.array([[1.], [-20], [100.]])

for i in range(1000):
    session.run(train, feed_dict={x: coefficients})
print(session.run(w))

4.9999886

三、重温代码

import numpy as np
import tensorflow as tf
coefficients = np.array([[1], [-20], [25]])
w = tf.Variable([0], dtype=tf.float32)
x = tf.placeholder(tf.float32, [3, 1])
cost = x[0][0]*w**2 + x[1][0] * w + x[2][0]
train = tf.train.GradientDescentOptimizer(0.01).minimize(cost)
init = tf.global_variables_initializer()
session.run(init)
print(session.run(w))

for i in range(1000):
    session.run(train, feed_dict={x: coefficients})
print(session.run(w))

[0.]
[9.999977]