add for in dict support (#22006)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/22006
ghimport-source-id: d9686c0b61b0eea3787f48adce567249e4e8faf0
Test Plan: Imported from OSS
Differential Revision: D15948548
Pulled By: wanchaol
fbshipit-source-id: 4227502ca050099085ad481aef725ac2cab06d74