Tensorflow中檢視權重

2021-08-28 12:03:44 字數 869 閱讀 2407

剛開始學習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檢視權重引數值

每個框架都有檢視權重引數的機制,在tenstensorflow中 檢視的例子如下 import tensorflow as tf import numpy as np reader tf.train.newcheckpointreader model 100 all variables reader...

Tensorflow 權重衰減的使用

在 tf.get variable 這個函式中有乙個命名引數為 regularizer,顧名思義,這個引數可用於正則化。在 tensorflow 官網中,regularizer 的描述如下 get variable name,shape none,dtype none,initializer non...

tensorflow 檢視ckpt模型中引數值

有時我們有檢視tensor內部變數的值的變化情況,需要挖一下,現給出解析 date 2021 5 18 author import osfrom tensorflow.python import pywrap tensorflow base model dir r model ckp 2021022...