https://obedmacallums.com/posts/advanced-techniques-in-pytorch/