tensorflow dataset使用demo
import tensorflow as tf
input_data = [1,2,3,5,8]
dataset = tf.data.Dataset.from_tensor_slices(input_data)
dataset = dataset.shuffle(4).repeat(2).batch(2)
# shuffle 随机打乱
# repeat 重复几次,相当于能够运行几个epoch
# batch 每次输出几个数据
iterator = dataset.make_one_shot_iterator()
x = iterator.get_next()
with tf.Session() as sess:
try:
for i in range(100):
print(sess.run(x))
except:
print('error')
finally:
print('done')
本文标题:tensorflow dataset使用demo
本文链接:https://www.haomeiwen.com/subject/qxondctx.html
网友评论