It easy to do k-Fold Cross Validation by using scikit-learn package. There we separately use data and labels. However, here combine data and labels before feeding into model. like:

 tf.data.Dataset.from_tensor_slices((X, Y))
        .shuffle(batch_size * 100)
        .batch(batch_size)

In this case how to do K-fold cross validation?