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