Reshape error

Expected Output:

Shape of training set after reshaping: (60000, 28, 28, 1)

Shape of one image after reshaping: (28, 28, 1)

Actual output:

Shape of training set after reshaping: (60000, 28, 28)

Shape of one image after reshaping: (28, 28)

[snippet removed by mentor]

you can see the full code error here, please guide me where i am making mistake.

[snippet removed by mentor]

I got the solution, in reshape_and_normalize function, for normalizing images, I should have put

[snippet removed by mentor]