你即將離開本站

並前往https://www.5axxw.com/questions/content/u1ewoz

如何用topk的索引（2D）得到topk的值？

pytorch的topk函数将提供以下信息。 top_tensor, indices = torch.topk(a, 2, dim=1) # top_tensor: tensor([[[3] ...

確定！ 回上一頁

查詢 「Torch topk」的人也找了：

  1. Torch topk
    2.
  2. Torch topk example
    3.
  3. Torch sort
    4.
  4. Pytorch topk 2d
    5.
  5. Pytorch topk gradient
    6.
  6. Torch eq
    7.
  7. Torch index_select
    8.
  8. Torch take
    9.