tensorflow缓存读入
```
tf.data.Dataset.from_tensor_slices((x_train, y_train))
train_dataset.shuffle(buffer_size=1024).batch(batch_size)
```
```
tf.data.Dataset.from_tensor_slices((x_train, y_train))
train_dataset.shuffle(buffer_size=1024).batch(batch_size)
```