tensorflow 数据集对象（tf.data）的使用( tf.data.Dataset 、tf.data.TextLineDataset 、 tf.data.TFRecordDataset ) 示例

tensorflow   使用数据集（tf.data）的方法对数据集进行操纵。

1.    对   数组（内存向量）  进行操纵 ：

import tensorflow as tf

input_data = [1, 2, 3, 4, 5]

#从数组生成数据集
dataset = tf.data.Dataset.from_tensor_slices(input_data)

#dataset = dataset.shuffle(3)
#dataset = dataset.repeat(10)
#dataset = dataset.batch(2)
dataset = dataset.shuffle(3).repeat(10).batch(2)


# 定义迭代器。
iterator = dataset.make_one_shot_iterator()

# get_next() 返回代表一个输入数据的张量(batch)。
x = iterator.get_next()
y = x * x


coord=tf.train.Coordinator()
with tf.Session() as sess:
    for i in range(25):
        print(sess.run(y))