tf中的计算图流程

最新推荐文章于 2022-01-14 18:20:10 发布

原创最新推荐文章于 2022-01-14 18:20:10 发布 · 709 阅读

0 ·

CC 4.0 BY-SA版权

tensorflow 专栏收录该内容

31 篇文章

订阅专栏

博客主要围绕TensorFlow中的计算图流程展开，但具体内容缺失。计算图在TensorFlow中是重要概念，了解其流程有助于更好地运用该框架进行开发。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#!/usr/bin/env python
# encoding: utf-8
'''
@author: lele Ye
@contact: 1750112338@qq.com
@software: pycharm 2018.2
@file: 10tf中的计算图流程
@time: 2018/12/16 10:36
@desc:
'''
import tensorflow as tf
# 定义一个变量state，初始值赋值为1.0
state = tf.Variable(1.0, dtype=tf.float32, name='counter')
# 定义一个常量 one
one = tf.constant(1.0, dtype=tf.float32, name="const")
# 将state与one相加赋值给new_value
new_value = tf.add(state, one)
# 通过tf.assign()修改变量state的值
update = tf.assign(state, new_value)

with tf.Session() as sess:
    # 如果定义了 variable必须进行初始化
    sess.run(tf.global_variables_initializer())
    for _ in range(3):
        print(sess.run(update))
        '''
        2.0
        3.0
        4.0
        '''