tensorflow的基本用法

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

#!/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)

执行结果如下图:

counter:0
1
2
3

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券