小编dar*_*ffy的帖子

我一直在尝试学习张量运算，而这个让我陷入了困境。
假设我有一个张量 t：

    t = torch.tensor([
        [1,0,0,2],
        [0,3,3,0],
        [4,0,0,5]
    ], dtype  = torch.float32)

现在这是一个 2 阶张量，我们可以对每个阶/维度应用 argmax。假设我们将其应用于 dim = 1

t.max(dim = 1)
(tensor([2., 3., 5.]), tensor([3, 2, 3]))

现在我们可以看到结果正如预期的那样，沿 dim =1 的张量有 2,3 和 5 作为最大元素。但是3上有冲突。有两个值完全相似。
怎么解决的？是任意选择的吗？有没有像LR一样的选择顺序，指数值越高？
如果您能了解如何解决此问题，我将不胜感激！