how to calculate Mean Rank in metric.py

[BodyCSoulN]

Hello,

I wanna know how to calculate Mean Rank in metric.py like MRR

is this right?
((indices == ground_truth_idx.cuda()).nonzero()[:, 1].float().add(1.0)).sum().item()