Pytorch: Real Step by Step implementation of CNN on MNIST