tensorflow迭代器

代碼:
import tensorflow as tf

dataset = tf.data.Dataset.range(100)
iterator = dataset.make_one_shot_iterator()
next_element = iterator.get_next()

with tf.Session() as sess:
print(sess.run(next_element))
print(sess.run(next_element))
print(sess.run(next_element))

iterator.get_next()可理解爲:把指針往後移動一個數據單位,再解除引用,返回相應數據
注意最開始,指針指向數據區域的前面,執行iterator.get_next()才指向第一個數據

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章