威尔tf.global_variables_初始值设定项()还初始化的迭代器tf.data.Dataset?

2024-09-28 23:45:53 发布

您现在位置:Python中文网/ 问答频道 /正文

我想知道tf.global_variables_initializer()是否也为tf.data.Dataset初始化{},或者我需要分别将{}初始化为:

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


Tags: rundatamaketfvariablesglobaldatasetsess
1条回答
网友
1楼 · 发布于 2024-09-28 23:45:53

必须分别初始化迭代器。 对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

相关问题 更多 >