ValueError: Found input variables with inconsistent numbers of samples: [32, 624]

Question

when plotting confusion matrix i got this error , if i set batch size is 624 and steps = 1 then code runs fine but if i set batch size 32 and steps 20 then i got this error