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?