K-Folds and TF.Datasets

Hello everyone,

I’m search for make k-folds or stratified k-folds with Tf.Datasets. It’s possible?

I don´t knowledge all tools of TF.Dataset, but use in some basics functions, maybe exist some function with help me to do this.

Thanks.