Added keep_key option to Grouper (#92532)
Fixes https://github.com/pytorch/data/issues/256
The testing of this module is currently suboptimal in general. We should improve this in the future.
@ejguan
Pull Request resolved: https://github.com/pytorch/pytorch/pull/92532
Approved by: https://github.com/ejguan