[PyTorch] Add an API restore from function context to ensure tensors are detached#2772

ksivaman merged 2 commits into

NVIDIA:mainNVIDIA/TransformerEngine:mainfrom

kainzhong:fix/change_restore_apikainzhong/TransformerEngine:fix/change_restore_apiCopy head branch name to clipboard