1. 程式人生 > >Tensorflow實戰-計算圖

Tensorflow實戰-計算圖

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

g1 = tf.Graph()
with g1.as_default():
    #在計算圖g1中定義變數“v”,並設定初始值為0.
    v = tf.get_variable(
        "v",initializer=tf.zeros_initializer()(shape=[1]) #維度(shape)資訊
    )

g2 = tf.Graph()
with g2.as_default():
    #在計算圖g2中定義變數“v”,並設定初始值為1.
    v = tf.get_variable(
        "v",initializer=tf.ones_initializer()(shape=[1])
    )

#在計算圖g1中讀取變數“v”的取值。
with tf.Session(graph=g1) as sess:
    tf.global_variables_initializer().run()
    with tf.variable_scope("",reuse=True): #生成一個名稱空間
        #在計算圖g1中,變數“v”的取值應該為0,所以下面這行會輸出[0.]。
        print (sess.run(tf.get_variable("v")))

#在計算圖g2中讀取變數“v”的取值。
with tf.Session(graph=g2) as sess:
    tf.global_variables_initializer().run()
    with tf.variable_scope("",reuse=True):
        #在計算圖g2中,變數“v”的取值應該為1,所以下面這行會輸出[1.]。
        print (sess.run(tf.get_variable("v")))