《tensorflow实战google》第二章代码

TensorFlow变量作用域示例

最新推荐文章于 2020-11-18 19:43:31 发布

原创最新推荐文章于 2020-11-18 19:43:31 发布 · 418 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow

机器学习专栏收录该内容

64 篇文章

订阅专栏

本文通过两个具体示例展示了如何在TensorFlow中使用不同的图来管理变量，并利用变量作用域来重用已创建的变量。首先在一个图中初始化了一个形状为[3]且初始值全为0的变量v，接着在另一个图中初始化了形状为[5]且初始值全为1的变量v。最后分别在这两个图中打印出了变量v的值。

部署运行你感兴趣的模型镜像

import tensorflow as tf

g1 = tf.Graph()
with g1.as_default():
    v = tf.get_variable("v", initializer=tf.zeros_initializer(), shape=[3])

g2 = tf.Graph()
with g2.as_default():
    v = tf.get_variable("v", initializer=tf.ones_initializer(), shape=[5])

with tf.Session(graph=g1) as sess:
    tf.initialize_all_variables().run()
    with tf.variable_scope("", reuse=True):
        print(sess.run(tf.get_variable("v")))

with tf.Session(graph=g2) as sess:
    tf.initialize_all_variables().run()
    with tf.variable_scope("", reuse=True):
        print(sess.run(tf.get_variable("v")))