How to split a whole tf.data.Dataset into images and labels?

Question

I\'m trying to split a separate distinct dataset into images and its labels to check my model against it.

I got the idea of just equating images, labels