搭建解决三好学生成绩问题的神经网络04---用向量重新组织输入数据

# Author:北京
# QQ:838262020
# time:2019/9/13
import tensorflow as tf

# 用3维的向量存入变量x
x = tf.placeholder(shape=[3], dtype=tf.float32)
yTrain = tf.placeholder(shape=[], dtype=tf.float32)

# 用3维向量存入可变参数w
w = tf.Variable(tf.zeros([3]), dtype=tf.float32)

n = w * x

# 所有维度的值相加求和，等于原来的y= n1+n2+n3
y = tf.reduce_sum(n)

loss = tf.abs(y - yTrain)

optimzer = tf.train.RMSPropOptimizer(0.001)

train = optimzer.minimize(loss)

sess = tf.Session()

init = tf.global_variables_initializer()

sess.run(init)

result = sess.run([train, x, w, y, yTrain, loss], feed_dict={x: [90, 80, 70], yTrain: 85})
print(result)

                                                                                                                   来源于《深度学习基于Python语言和TensorFlow平台》