ValueError: Shape mismatch: The shape of labels (received (320,)) should equal the shape of logits except for the last dimension (received (32, 2))

Question

I\'m a beginner. I\'m stuck at this test step. My code is running well. I\'m getting around 91% accuracy but when I try to test it. It shows an error-> ValueError: Shape