Updates #836 as suggested in https://github.com/pytorch/pytorch/issue…

[TheMemoryDealer]

…s/16885#issuecomment-551779897

Fixes #836

Description

Fixes issue discussed in #16885

Checklist

• The issue that is being fixed is referred in the description (see above "Fixes #ISSUE_NUMBER")
• Only one issue is addressed in this pull request
• Labels from the issue that this PR is fixing are added to this pull request
• No unnessessary issues are included into this pull request.

[netlify]

✅ Deploy Preview for pytorch-tutorials-preview ready!

Name	Link
🔨 Latest commit	4b6497c
🔍 Latest deploy log	https://app.netlify.com/sites/pytorch-tutorials-preview/deploys/64787f7daf35750008f8ac70
😎 Deploy Preview	https://deploy-preview-2388--pytorch-tutorials-preview.netlify.app/beginner/former_torchies/parallelism_tutorial
📱 Preview on mobile	Toggle QR Code... Use your smartphone camera to open QR code link.

---

To edit notification comments on pull requests, go to your Netlify site settings.