Modernize for-loops (#50899)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50899
Test Plan: Sandcastle tests + OSS CI
Reviewed By: ezyang
Differential Revision: D26001931
fbshipit-source-id: d829d520f647aacd178e1c7a9faa6196cc5af54e
