pytorch

d9117b93 - unsqueeze only when dim = 3 (#91052)

unsqueeze only when dim = 3 (#91052) unsqueeze is not necessary if use view Pull Request resolved: https://github.com/pytorch/pytorch/pull/91052 Approved by: https://github.com/albanD