Update device mesh tutorial to use correct API#3525

SwayamInSync wants to merge 3 commits intopytorch:mainpytorch/tutorials:mainfrom

SwayamInSync:mainSwayamInSync/tutorials:mainCopy head branch name to clipboard