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