Update the CNN notebook using the latest TensorFlow version

I want to update [this](https://github.com/donnemartin/data-science-ipython-notebooks/blob/master/deep-learning/tensor-flow-examples/notebooks/3_neural_networks/convolutional_network.ipynb) notebook using the latest TensorFlow version 2.4.1.

Please assign this issue to me.