你即將離開本站

並前往https://www.coder.work/article/4986844

python - 为什么dim = 1返回torch.argmax中的行索引？ - IT工具网

我正在使用PyTorch的 argmax 功能定义为： torch.argmax(input, dim=None, keepdim=False) 考虑一个例子 a = torch.randn(4, 4) print(a) print(torch.argmax(a, dim=1))

確定！ 回上一頁

查詢 「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.