tensorflow的基本用法(三)——Variable

文章作者:Tyan
博客:noahsnail.com  |  CSDN  |  简书

本文主要是介绍tensorflow中的变量Variable及用法。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
#!/usr/bin/env python
# _*_ coding: utf-8 _*_

import tensorflow as tf
import numpy as np

# 定义变量
state = tf.Variable(0, name = 'counter')
print(state.name)

# 定义常量
one = tf.constant(1)
# 定义新的变量,值等于state + one
new_value = tf.add(state, one)
# 定义update操作,将new_value赋值给state
update = tf.assign(state, new_value)

# 定义的变量激活,如果定义了Variable,必须有
init = tf.global_variables_initializer()

with tf.Session() as sess:
sess.run(init)
for _ in range(3):
# 执行update操作,update操作包括add,assign两部分
sess.run(update)
# 输出state结果,必须放到sess.run()中
print sess.run(state)

执行结果如下图:

1
2
3
4
counter:0
1
2
3

参考资料

  1. https://www.youtube.com/user/MorvanZhou
如果有收获,可以请我喝杯咖啡!