你即將離開本站

並前往https://qa.1r1g.com/sf/ask/3898427361/

为什么dim=1 在torch.argmax 中返回行索引？ | 经验摘录

在这里，当我使用dim=1 而不是搜索列向量时，该函数会搜索行向量，如下所示。 print(a) : tensor([[-1.7739 ...

確定！ 回上一頁

查詢 「Torchargmax」的人也找了：

  1. Pytorch maximum
    2.
  2. Tensor to int
    3.
  3. .item pytorch
    4.
  4. Torch topk
    5.
  5. Torch expand
    6.
  6. Torch .max outputs.data, 1
    7.
  7. Torch argsort
    8.
  8. Pytorch masked_fill
    9.