I’m new to Torch, and there’s one thing I find counterintuitive in argmin and argmax operations,
so, per example from the docs we have
>>> a = torch.randn(4, 4)
>>> a
tensor([[ 0.1139, 0.2254, -0.1381, 0.3687],
[ 1.0100, -1.1975, -0.0102, -0.4732],
[-0.9240, 0.1207, -0.7506, -1.0213],
[ 1.7809, -1.2960, 0.9384, 0.1438]])
>>> torch.argmin(a)
tensor(13)
But wouldn’t it be more natural to return tensor([3,1])? In particular case, I have two distinct sets of points A and B, and I look for the indices of a pair (a,b) ∈ A × B. I can measure pairwise distances using cdist, but a 42-like answer is of little use. Sure the indices can be inferred, but doing so looks messy and superfluous.
Is there any better solution? And also, are there any reason for delivering argmin/argmax this way beyond backward compatibility and seeking in an effectively linear array?