最近学习在 TensorFlow,看了两行代码后觉得TensorFlow真是一个不错的工具。然后,作为工科男我就想着解决一些有趣的问题;解决什么好呢?想来想去还是决定对我们熟悉的二次方程求解。我想大家对如何解二次方程早就成竹在胸,下面看一下TensorFlow如何做的吧。
问题:
我们先来看一下需要求解的方程:y = (x+1)^2;
然后给出我们的要求:求解在 y 最小时 x 的值。
那么如何求解呢?OK,不卖关子啦,直接上代码喽!
代码段:
# coding:utf-8
# 设函数 y = (x+1)^2, 令 x 初始值为5
# 反向传播就是求最优 x,即求最小y值对应的x值
import tensorflow as tf
# 定义待优化参数 x 初值为 5
x = tf.Variable(tf.constant(5, dtype=tf.float32)