pytorch

01006415 - Fix the issue described by #106532 (#108036)

Fix the issue described by #106532 (#108036) Fixes #106532 As the title says Pull Request resolved: https://github.com/pytorch/pytorch/pull/108036 Approved by: https://github.com/albanD