Dimension out of range (expected to be in range of [-1, 0], but got 1)

Question

I have been trying to implement MNIST dataset on a simple Neural Network(784,512,128,10) using cross-entropy loss. I am using Keras for getting the MNIST dataset. But I\'m getti