Modernize some loops in torch (#61737)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61737
Test Plan: Sandcastle
Reviewed By: malfet
Differential Revision: D29716813
fbshipit-source-id: 21f9716bead4e0e913406e681c55d1956327e6af
