fast setup for output tensor in tensor iterator (#33165)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/33165
Test Plan: Imported from OSS
Differential Revision: D19825853
Pulled By: glaringlee
fbshipit-source-id: 8f908f2e93a4e377306a77e8a771208603b20e72
