你即將離開本站

並前往https://runebook.dev/zh-CN/docs/pytorch/generated/torch.topk

torch.topk - 返回沿给定维度的给定input 张量的k 个最大元素 ...

如果未指定dim ，则选择input 的最后一个尺寸。 如果largest 为False ,则返回k 个最小元素。 返回(values, indices) namedtuple ，其中indices 是原始input 张量中元素 ...

確定！ 回上一頁

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