ValueError: zero-dimensional arrays cannot be concatenated

[ArshadIram]

Thank you so much for sharing the code for reproducibility.

I am getting the error because idx is ! greater 0.

idx = np.where(label.cpu().numpy() == np.array([0]*data.shape[0]))[0]

It will be great if you help me in resolving the issue.