代码之家 › 专栏 › 技术社区 › Panfeng Li

tf.global_variables_initializer()是否也会初始化tf.data.dataset的迭代器?

tensorflow-datasets tensorflow python

0

Panfeng Li · 技术社区 · 7 年前

我想知道 tf.global_variables_initializer() 同时初始化 iterator 对于 tf.data.Dataset 或者我需要初始化 迭代器 分别为:

iterator = dataset.make_initializable_iterator() sess.run(iterator.initializer)

1 回复 | 直到 7 年前

1

2

BugKiller 7 年前

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