confusion in OhemCrossEntropy loss function

[Srutsth]

pred = prob[label, np.arange(len(label), dtype=np.int32)]

What does this line do in OhemCrossEntropy do?

[Serge-weihao]

CCNet-Pure-Pytorch/loss/loss.py

Line 68 in ff51a27

pred = prob[label, np.arange(len(label), dtype=np.int32)]

I guess that means to get the gt class related predicted score for each (valid) pixel.