Got error in Reshaping Images In CNN( Googel Colab)

Question

I have a CNN project of digit recognization. I am trying it in colab. I have a 1D vector with 784 pixels and I have to reshape it to (28x28x1) before passing it to the CNN.