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.