[feat] exposed customed api-pytorch-AtomGit

Star857

Star857

已合并

[feat] exposed customed api #31388

culechan创建于 3月4日从culechan/pytorch： cherry-pick-mr-31387-1772627906510-auto 合入到 Ascend/pytorch： v2.9.0

[feat] exposed customed api #31388

已合并

culechan创建于 3月4日

从culechan/pytorch cherry-pick-mr-31387-1772627906510-auto合入到Ascend/pytorch v2.9.0