Sorry for a late reply.
Unfortunately, there is no a simple way but to count it manually:
num = 0 for batch in stream.get_epoch_iterator(): num += 1 print(num)