你即將離開本站

並前往https://stackoverflow.com/questions/60434505/why-is-torchnonzero-very-slow

why is torch::nonzero() very slow? - Stack Overflow

Your code is not wrong. torch::nonzero has to check every entry in the tensor you give it, in this case 30000*80=2.4e+6 .

確定！ 回上一頁

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

  1. Torch nonzero
    2.
  2. Torch nonzero 用法
    3.
  3. Torchtensor
    4.
  4. Torch first non-zero
    5.
  5. Torch nonzero gradient
    6.
  6. Numpy nonzero
    7.
  7. Pytorch count nonzero
    8.
  8. Pytorch find
    9.