May I ask how to do masked argmax in Pytorch?
For example I have a tensor t and a mask m:
t = torch.tensor([20, 10, 50, 40])
m = numpy.array([True, True, False, True])
The masked argmax is 3 (corresponds to 40 in t)
Thank you very much.