Source generation skips when input batch size exceeds source length

[amitschang]

For example if a generator stops iteration at 9 batches but the input batch size is 10, this will yield nothing.