__author__ = 'Administrator' import os os.environ['TF_CPP_MIN_LOG_LEVEL']='2' import tensorflow as tf import numpy as np x_data = np.float32(np.random.rand(2, 100)) y_data = np.dot([0.100, 0.200], x_data) + 0.300 b = tf.Variable(tf.zeros([1])) W = tf.Variable(tf.random_uniform([1, 2],-1.0, 1.0)) y = tf.matmul(W, x_data) + b loss = tf.reduce_mean(tf.square(y - y_data)) optimizer = tf.train.GradientDescentOptimizer(0.5) train = optimizer.minimize(loss) init = tf.initialize_all_variables() sess = tf.Session() sess.run(init) for step in range(0, 101): sess.run(train) if step % 10 == 0: print(step, sess.run(W), sess.run(b))
如果输出:
0 [[-0.21173605 0.8310676 ]] [0.29118115]
10 [[-0.00951169 0.45860067]] [0.21995221]
20 [[0.07681578 0.32292184]] [0.2460896]
30 [[0.10260113 0.26202923]] [0.26505017]
40 [[0.10803779 0.23306021]] [0.2777703]
50 [[0.10741194 0.21843833]] [0.28601575]
60 [[0.10551126 0.21064365]] [0.29125994]
70 [[0.10376108 0.20629568]] [0.29455885]
80 [[0.10246037 0.20378552]] [0.29662058]
90 [[0.10157292 0.20230065]] [0.2979041]
100 [[0.10099245 0.20140775]] [0.29870132]
证明tensorflow安装没有任何问题。