Code snippets created for the PyTorch discussion board - View it on GitHub