Un fold squeeze permute (#91656)
Fixes #91505
Hey, this should partially fix some of the problems discussed in the issue above-if I'm on the right track, I'll update this PR with more fixes.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/91656
Approved by: https://github.com/ngimel
