https://camera-optical-zoom.blogspot.com/2023/12/deriving-structure-of-pytorch-network.html