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: 数据重复训练次数