tensorflow入门（一）--计算图的生成

最新推荐文章于 2024-08-16 17:33:04 发布

原创最新推荐文章于 2024-08-16 17:33:04 发布 · 533 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #tensorflow #入门

本文介绍TensorFlow中计算图的创建与使用，包括如何定义常量、变量，以及通过会话运行计算图的基本操作。展示了如何在计算图中进行简单的数学运算，并初始化和读取变量。

import tensorflow as tf
g1=tf.Graph()   #建立计算图，tensorflow里有默认的，也可以自己建一个
with g1.as_default():
    a=tf.constant([1,2,3],name='a')  #常量
    c=a+1
    b=tf.get_variable('b',initializer=tf.constant_initializer()(shape=[4]))#变量
with tf.Session(graph=g1) as sess:
    #with g1.device("/gpu:0"):   #在gpu上运行，但需要先设置
    tf.global_variables_initializer().run()   #初始化所有变量，如果有variable，必须要先初始化
    print(sess.run(c))
    print(sess.run(b))