https://catalogcasiokeyboardlk.blogspot.com/2024/05/deriving-structure-of-pytorch-network.html