Add ShuffleDataset with buffer (#45290)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45290
Test Plan: Imported from OSS
Reviewed By: gchanan
Differential Revision: D24001084
Pulled By: erjia-guan
fbshipit-source-id: d8a7455cf3f18e1f8c1edc53c42c1a99c8573c51
