美文网首页
tensorflow dataset使用demo

tensorflow dataset使用demo

作者: snowpigppp | 来源:发表于2019-08-07 10:40 被阅读0次
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