python - tf.global_variables_initializer() 还会初始化 tf.data.Dataset 的迭代器吗？

Question

我想知道tf.global_variables_initializer()是否也初始化tf.data.Dataset的iterator，或者我需要初始化迭代器分别为:


迭代器 = dataset.make_initialized_iterator()
sess.run(迭代器.初始化器)


Answer 1

您必须单独初始化迭代器。tf.global_variables_initializer()

没有 无变量提要

official example :

max_value = tf.placeholder(tf.int64, shape=[])
dataset = tf.data.Dataset.range(max_value)
iterator = dataset.make_initializable_iterator()
next_element = iterator.get_next()

# Initialize an iterator over a dataset with 10 elements.
sess.run(iterator.initializer, feed_dict={max_value: 10})
for i in range(10):
  value = sess.run(next_element)
  assert i == value

# Initialize the same iterator over a dataset with 100 elements.
sess.run(iterator.initializer, feed_dict={max_value: 100})
for i in range(100):
  value = sess.run(next_element)
  assert i == value