根據(jù)tensor的名字獲取變量的值方式
需求:
有時候使用slim這種封裝好的工具,或者是在做滑動平均時,系統(tǒng)會幫你自動建立一些變量,但是這些變量只有名字,而沒有顯式的變量名,所以這個時候我們需要使用那個名字來獲取其對應(yīng)的值。
如下:
input = np.random.randn(4,3) net = slim.fully_connected(input,2,weights_initializer=tf.ones_initializer(dtype = tf.float32))
這段代碼看似簡單,但其實幫你生成了一個w和一個b。如果你運行下面代碼:
with tf.Session() as sess: sess.run(tf.global_variables_initializer()) for v in tf.global_variables(): print(v)
你會發(fā)現(xiàn)里面還有
<tf.Variable 'fully_connected/weights:0' shape=(3, 2) dtype=float64_ref> <tf.Variable 'fully_connected/biases:0' shape=(2,) dtype=float64_ref>
這樣兩個變量,但是由于沒有顯式聲明,所以我們要從其名字取值。
解決方案:
1、從圖里面取值:
print(sess.run(tf.get_default_graph().get_tensor_by_name("fully_connected/weights:0")))
這個就是先拿到圖,然后從圖里面取變量 。
2、直接取值
print(sess.run("fully_connected/weights:0"))
直接把名字傳進run里面就可以直接運行了,但是這個仍然拿不到變量,這個只能拿到變量值。
以上這篇根據(jù)tensor的名字獲取變量的值方式就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
python?動態(tài)規(guī)劃問題解析(背包問題和最長公共子串)
這篇文章主要介紹了python?動態(tài)規(guī)劃(背包問題和最長公共子串),在動態(tài)規(guī)劃中,你要將某個指標(biāo)最大化。在這個例子中,你要找出兩個單詞的最長公共子串。fish和fosh都包含的最長子串是什么呢,感興趣的朋友跟隨小編一起看看吧2022-05-05pytorch載入預(yù)訓(xùn)練模型后,實現(xiàn)訓(xùn)練指定層
今天小編就為大家分享一篇pytorch載入預(yù)訓(xùn)練模型后,實現(xiàn)訓(xùn)練指定層,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01Python實現(xiàn)json對值進行模糊搜索的示例詳解
我經(jīng)常使用json進行存儲配置,于是常常遇到這樣的問題:如果想要對某個數(shù)組里的值進行模糊搜索,同時輸出相關(guān)的其他數(shù)組相同位置的的值該如何實現(xiàn)呢?本文就來和大家詳細聊聊2023-01-01