[LeNet] [Conv2d] * K: kernel size = 5 * S: stride = 2 #0 Input RGB image = 1, 28, 28 #1 Conv2d + ReLU = 32, 14, 14 #2 Conv2d + ReLU = 64, 7, 7 #3 Transform = 64 * 7 * 7 #4 Linear + Sigmoid = 10