https://obedmacallums.com/posts/getting-started-with-pytorch/