Revert "Handle 2D blocking with foreach (#107840)"
This reverts commit f87ffe473d4825c15eaea0360baf08cad49979de.
Reverted https://github.com/pytorch/pytorch/pull/107840 on behalf of https://github.com/huydhn due to Sorry for reverting this, but test_2d_blocking is failing in trunk, probably a landrace as PR was green ([comment](https://github.com/pytorch/pytorch/pull/107840#issuecomment-1694009217))