Invalid Argument Error Logits and Shapes ( Keras, Tensorflow)

Question

Running in google Colab

Here is the necessary code. I just want it to work, doesn\'t even need to be high accuracy.

#Generator, different from function