convolution and pooling performed in PyTorch along with classification using two conv + maxPool layer on MNIST dataset