change input shape to reduce variation (#27548)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/27548
as title
Test Plan: i_dont_want_it
Reviewed By: hl475
Differential Revision: D17811295
fbshipit-source-id: 3be957f6f3eaa464ebf4f5bd7c07d096ae4eae8c