tf.data.Dataset.from_tensor_slices( ) 解释
使用这个函数转化数据集时,按照传入数据的第一维度切分,生成相应的 dataset 。
ds = tf.data.Dataset.from_tensor_slices((x, y)) ## 输入 x,y 进行切分 一一对应
train_ds = ds.take(20000).shuffle(20000).batch(100)
take : 创建一个元素集合,最多包含多少个元素
shuffle: 打乱元素
batch: 选取多少个元素作为一个batch大小
repeat: 数据重复训练次数