Tensorflow generator for a certain subset

Each time I evaluate dataset I just want to evaluate an exact subset of the dataset that I already have index. I use a generator iterator to read the input, but everytime I run it always read more input than expect so sometimes the input from last run mixed with the input for the next run, which is a serious problem. Anyone know anyway to read exactly a subset of input with generator?

Hi @Dung_Hoang_Anh

Welcome to the TensorFlow Forum!

Could you please share minimal reproducible code to replicate and understand the issue better? Thank you.