tensorflow 中的dataset是一个基类,可以用来处理数据的输入。
数据生成
可以用以下方法生成dataset:
- from_tensor_slices()
将输入的张量、元组、字典进行切片处理。切片发生在最外层的维度。比如将(5,2)形状的张量输入,得到的是5个(2)形状的张量:
import tensorflow as tf
s = tf.constant(range(1,11))
s = tf.reshape(s,(5,2))
t = tf.data.Dataset.from_tensor_slices(s)
print(list(t.as_numpy_iterator()))
结果:
[array([1, 2]), array([3, 4]), array([5, 6]), array([7, 8]), array([ 9, 10])]
对于有监督的学习,该函数还可以生成(特征,