IterKeyZipper has an option to keep the key that was zipped on:

https://github.com/pytorch/data/blob/2cf1f208e76301f3e013b7569df0d75275f1aaee/torchdata/datapipes/iter/util/combining.py#L53

Is this something we want to support going forward? If yes, it would be nice to have this also on Grouper and possibly other similar datapipes. That would come in handy in situations if the key is used multiple times for example if we have a IterKeyZipper after an Grouper.

Additional Context for New Contributors

See comment below