Example

dataset = tf.data.Dataset.from_tensor_slices(DATA).repeat().batch(BATCH_SIZE)
iterator = dataset.make_initializable_iterator()
next_element = iterator.get_next()

sess = tf.Session()
sess.run(iterator.initializer)
for i in range(ITERATION):
training_batch = sess.run(next_element)

Reference

http://adventuresinmachinelearning.com/tensorflow-dataset-tutorial/
https://towardsdatascience.com/how-to-use-dataset-in-tensorflow-c758ef9e4428