How to store wrong predictions during evaluation on the CNN

Question

during evaluation I want to store unique ids that are wrongly predicted to do some more processing. It is a multiclass prediction problem

here is the code during the