reland map/zip iterator related changes (#135074)
Summary:
X-link: https://github.com/pytorch/pytorch/pull/135074
Approved by: https://github.com/jansel, https://github.com/anijain2305, https://github.com/mlazos
Reviewed By: kit1980
Differential Revision: D62211019
Pulled By: williamwen42
fbshipit-source-id: 3e4cf12548bb95b66ee7272a6d77ebffa93b6046
