tensorflow如何初始化未初始化的變量

這篇文章將為大家詳細(xì)講解有關(guān)tensorflow如何初始化未初始化的變量，小編覺(jué)得挺實(shí)用的，因此分享給大家做個(gè)參考，希望大家閱讀完這篇文章后可以有所收獲。

今日在Stack Overflow上看到一個(gè)問(wèn)如何只初始化未初始化的變量，有人提供了一個(gè)函數(shù)，特地粘貼過(guò)來(lái)共大家品鑒：

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))

通過(guò)tf.global_variables()返回一個(gè)全局變量的列表global_vars, 然后以python列表解析式的方式判斷全局變量列表global_vars中每個(gè)變量是否都被初始化了，并返回一個(gè)對(duì)應(yīng)的True/False的布爾值列表is_not_initialized；

然后再把global_vars中的變量和布爾值列表進(jìn)行打包配對(duì)，并選出沒(méi)有初始化的變量的列表，得到所有未初始化變量的列表not_initialized_vars；打印出這些變量的名字；

對(duì)于這些未初始化的變量，通過(guò)會(huì)話（sess）運(yùn)行tf.variables_initializer(not_initialized_vars)使所有未初始化的變量都初始化。

關(guān)于“tensorflow如何初始化未初始化的變量”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，使各位可以學(xué)到更多知識(shí)，如果覺(jué)得文章不錯(cuò)，請(qǐng)把它分享出去讓更多的人看到。