Vous pouvez le convertir en liste avec list(ds) puis recompilez-le comme un jeu de données normal avec tf.data.Dataset.from_tensor_slices(list(ds)) . A partir de là ton cauchemar recommence mais au moins c'est un cauchemar que d'autres personnes ont déjà fait.
Notez que pour les ensembles de données plus complexes (par exemple, les dictionnaires imbriqués), vous aurez besoin de plus de prétraitement après avoir appelé list(ds) , mais cela devrait fonctionner pour l'exemple que vous avez demandé.
C'est loin d'être une réponse satisfaisante, mais malheureusement, la classe est entièrement non documentée et aucune des astuces standard de l'ensemble de données ne fonctionne.
