Creating small subsets on a dataset and training with different subsets for each epoch in TensorFlow 2

Question

Say I have a dataset of 5000 images. Rather than training my model on the same whole 5000 images, I want to train it on small subsets of 200 images on each epoch and want to