I need a 2 dimensional array to compute the ROC score, when I forward the model per batch it gives me a 2 dimension array but I need to collect all my batch predictions so I used list.append() that is turning the predictions to 3d hence I  can’t compute… Pls How do I store the batch predictions?