Update split iteration for DynamicBatchingDataset (#684)
This PR aims to make iterating over splits a bit more intuitive, at least in my opinion. Open to feedback though! If the current behavior was intentional, feel free to close.
---------
Co-authored-by: Nathan Habib <30601243+NathanHB@users.noreply.github.com>