你即將離開本站

並前往https://wizardforcel.gitbooks.io/pytorch-03-doc/torch.html

torch · PyTorch 0.3 中文文档

torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor). 沿给定dim 维度返回输入张量 input 中 k 个最大值.

確定！ 回上一頁

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