tensorflow缓存读入

  ```
tf.data.Dataset.from_tensor_slices((x_train, y_train))

train_dataset.shuffle(buffer_size=1024).batch(batch_size)

```