https://www.cheatsheetshero.com/user/yousef-yousefmejdi/1143-deep-learning-with-pytorch-cheatsheet