Python and ValueError: could not broadcast input array?

Question

I am getting the below error:

 could not broadcast input array from shape (128,784) into shape (128)

This is my code:

import n