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]