remove unsqueeze from test

Summary:
seems like decomposition has changes so we can loosen this test

https://github.com/pytorch/executorch/pull/2162

Reviewed By: kit1980, guangy10

Differential Revision: D54318979

fbshipit-source-id: 000c6e5a61fb1cb22f1dce92c669c6de61e66baf