Tensorflow没有正确计算权重?

时间:2017-10-09 15:30:32

标签: tensorflow

我在tensorflow中进行编码,其中输入等于输出,因此权重应为1且偏差为零。但它没有给出正确的结果

代码如下所示

import tensorflow as tf
x=tf.placeholder(tf.float32,shape=[None,1])
y_=tf.placeholder(tf.float32,shape=[None,1])
W=tf.Variable(tf.zeros([1,1]))
b=tf.Variable(tf.zeros([1]))
y=tf.matmul(x,W)+b
init=tf.global_variables_initializer()
cross_entropy=tf.nn.softmax_cross_entropy_with_logits(labels=y_,logits=y)
train_step=tf.train.GradientDescentOptimizer(0.6).minimize(cross_entropy)
sess=tf.InteractiveSession()
sess.run(init)
for e in range(100):
    sess.run([train_step,cross_entropy],feed_dict={x:[[1],[2]],y_:[[1],[2]]})
print(sess.run([W,b]))

0 个答案:

没有答案