tf.data.Dataset - delete cache?

Question

Is it possible to delete the in-memory cache that\'s built after calling tf.data.Dataset.cache()?

Here\'s what I\'d like to do. The augmentation for the