https://cheatsheetshero.com/user/all/446-pytorch-cheat-sheet