[quant] Fix dimension for output of batchnorm 1d (#59264)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/59264
Previously batchnorm 1d did unsqueeze twice but only squeeze once before return when the dimension
for input Tensor is 2, this PR adds an extra squeeze
Test Plan: Imported from OSS
Reviewed By: supriyar
Differential Revision: D28810597
fbshipit-source-id: 879873bbf39ed3607762684694f6e81b423740c2
