test: fix failed test cases in test_optim-pytorch-AtomGit

Star844

Star844

已合并

test: fix failed test cases in test_optim #35789

21xiu创建于 5月15日从xiu_21/pytorch： cherry-pick-mr-35788-1778837035527-auto 合入到 Ascend/pytorch： v2.10.0

test: fix failed test cases in test_optim #35789

已合并

21xiu创建于 5月15日

从xiu_21/pytorch cherry-pick-mr-35788-1778837035527-auto合入到Ascend/pytorch v2.10.0