新聞中心
這里有您想知道的互聯(lián)網(wǎng)營銷解決方案
tensorflow如何初始化未初始化的變量-創(chuàng)新互聯(lián)
這篇文章將為大家詳細(xì)講解有關(guān)tensorflow如何初始化未初始化的變量,小編覺得挺實用的,因此分享給大家做個參考,希望大家閱讀完這篇文章后可以有所收獲。
今日在Stack Overflow上看到一個問如何只初始化未初始化的變量,有人提供了一個函數(shù),特地粘貼過來共大家品鑒:
import tensorflow as tf def initialize_uninitialized(sess): global_vars = tf.global_variables() is_not_initialized = sess.run([tf.is_variable_initialized(var) for var in global_vars]) not_initialized_vars = [v for (v, f) in zip(global_vars, is_not_initialized) if not f] print [str(i.name) for i in not_initialized_vars] # only for testing if len(not_initialized_vars): sess.run(tf.variables_initializer(not_initialized_vars))
通過tf.global_variables()返回一個全局變量的列表global_vars, 然后以python列表解析式的方式判斷全局變量列表global_vars中每個變量是否都被初始化了,并返回一個對應(yīng)的True/False的布爾值列表is_not_initialized;
然后再把global_vars中的變量和布爾值列表進(jìn)行打包配對,并選出沒有初始化的變量的列表,得到所有未初始化變量的列表not_initialized_vars;打印出這些變量的名字;
對于這些未初始化的變量,通過會話(sess)運行tf.variables_initializer(not_initialized_vars)使所有未初始化的變量都初始化。
關(guān)于“tensorflow如何初始化未初始化的變量”這篇文章就分享到這里了,希望以上內(nèi)容可以對大家有一定的幫助,使各位可以學(xué)到更多知識,如果覺得文章不錯,請把它分享出去讓更多的人看到。
網(wǎng)頁題目:tensorflow如何初始化未初始化的變量-創(chuàng)新互聯(lián)
網(wǎng)頁路徑:http://www.dlmjj.cn/article/dpcipj.html