https://keroseneheaterreview.blogspot.com/2023/08/understanding-output-shape-of-conv2d.html