你即將離開本站

並前往http://www.4k8k.xyz/article/weixin_41449637/120090965

np.unravel_index 和torch.argmax() ——（标题党） - 程序员资料

np.unravel_index 返回多维矩阵中的最大值对应的索引，如二维，则返回最大值对应的行列号torch中没有unravel_index的函数，不过可以用torch.argmax实现， ...

確定！ 回上一頁

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