本节主要详述tensorflow的常量与变量,以下代码在python3.5,tensorflow1.5.0,numpy1.13.3下测试通过,想学习的小伙伴可以直接拷贝运行,一块学习提高呀。
1 用tensorflow实现计数器,主要是设计了在循环中调用加法实现计数
代码:
import tensorflow as tf
import numpy as np
# 创建变量,初始化为0
state = tf.Variable(0, name="counter")
# 创建一个 op , 其作用是时 state 增加 1
one = tf.constant(1) # 直接用 1 也就行了
new_value = tf.add(state, 1)
update = tf.assign(state, new_value)
# 启动图之后, 运行 update op
with tf.Session() as sess:
# 创建好图之后,变量必须经过‘初始化’
sess.run(tf.global_variables_initializer())
# 查看state的初始化值
print sess.run(state)
for _ in range(3):
sess.run(update) # 这样子每一次运行state 都还是1
print sess.run(state)
运行结果:
0
1
2
3
2 用 tf 来实现对一组数求和,再计算平均
代码:
h_sum = tf.Variable(0.0, dtype=tf.float32)
# h_vec = tf.random_normal(shape=([10]))
h_vec = tf.constant([1.0,2.0,3.0,4.0])
# 把 h_vec 的每个元素加到 h_sum 中,然后再除以 10 来计算平均值
# 待添加的数
h_add = tf.placeholder(tf.float32)
# 添加之后的值
h_new = tf.add(h_sum, h_add)
# 更新 h_new 的 op
update = tf.assign(h_sum, h_new)
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
# 查看原始值
print 's_sum =', sess.run(h_sum)
print "vec = ", sess.run(h_vec)
# 循环添加
for _ in range(4):
sess.run(update, feed_dict={h_add: sess.run(h_vec[_])})
print 'h_sum =', sess.run(h_sum)
# print 'the mean is ', sess.run(sess.run(h_sum) / 4) # 这样写 4 是错误的, 必须转为 tf 变量或者常量
print 'the mean is ', sess.run(sess.run(h_sum) / tf.constant(4.0))
运行结果:
s_sum = 0.0
vec = [ 1. 2. 3. 4.]
h_sum = 1.0
h_sum = 3.0
h_sum = 6.0
h_sum = 10.0
the mean is 2.5
如果代码有看不懂的小伙伴,可以留言或者私信博主(* ̄︶ ̄)。