Specs
Input Image is 28x28x1 and converted to 32x32x1 as per LeNet requirements.
Convolution layer 1: The output shape should be 28x28x6.
Activation 1: Your choice of activation function.
Pooling layer 1: The output shape should be 14x14x6.
Convolution layer 2: The output shape should be 10x10x16.
Activation 2: Your choice of activation function.
Pooling layer 2: The output shape should be 5x5x16.
Flatten layer: Flatten the output shape of the final pooling layer such that it's 1D instead of 3D.
Fully connected layer 1: This should have 120 outputs.
Activation 3: Your choice of activation function.
Fully connected layer 2: This should have 10 outputs.
Return the result of the 2nd fully connected layer from the LeNet function.