1. 程式人生 > 程式設計 >在Tensorflow中檢視權重的實現

在Tensorflow中檢視權重的實現

剛開始學習tensorflow,還不太會用,開個博記錄,今天遇到一個問題是用tf.layers.dense建立的全連線層,如何檢視權重?

知道kernel表示了權重,但是如何提示成變數?

我分成兩步:

1、檢視tensor:tf.trainable_variables()

命令列裡中執行即可,如下圖:

可以看到tensor的name,或用更簡潔的方式:variable_name = [v.name for v in tf.trainable_variables()]

執行後,trainable_variable()即存入variable_name中:

2、獲取權重:tf.get_default_graph().get_tensor_by_name('variable_name')

如:w1=tf.get_default_graph().get_tensor_by_name('Actor/l1/kernel:0')

即可根據get tensor by name

菜鳥的笨方法^-^

—————————————————————————————————

更新:tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES,scope)也可檢視,可得到某scope下的tensor

即從某scope的tf.GraphKeys.TRAINABLE_VARIABLES中取出所有變數,是一個列表

以上這篇在Tensorflow中檢視權重的實現就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。