TF通过feature与label生成(特征,标签)集合,tf.data.Dataset.from_tensor_slices

Numpy和Tensor格式都可以用这个语句读入数据

data = tf.data.Dataset.from_tensor_slices((输入特征, 标签))

代码

import tensorflow as tf

features = tf.constant([12, 23, 10, 17])
labels = tf.constant([0, 1, 1, 0])
dataset = tf.data.Dataset.from_tensor_slices((features, labels))
print(dataset)
for element in dataset:
    print(element)