Recall CE runtime error

[YFrisch]

Hi. This simple test with the Recall CE

gt = torch.randint(low=0, high=10, size=(8, 32, 32)).cuda()
pred = torch.rand(size=(8, 10, 32, 32)).cuda()
recal_ce = RecallCrossEntropy(n_classes=10)
L = recal_ce(pred, gt)

throws a runtime error

>       gt_count[gt_idx == self.ignore_index] = gt_count[1]
E       RuntimeError: unsupported operation: some elements of the input tensor and the written-to tensor refer to a single memory location. Please clone() the tensor before performing the operation.

Using Python 3.10