你即將離開本站

並前往https://discuss.pytorch.org/t/how-to-get-index-of-non-zero-values-for-every-row-in-a-matrix-pytorch/99450

How to get index of non-zero values for every row in a matrix ...

You can try torch.nonzero(x) , or x.ne(0) (meaning 'not equal'). One of the two solutions will probably suit your needs.

確定！ 回上一頁

查詢 「Torch first non-zero」的人也找了：

  1. Torch nonzero
    2.
  2. Torch where index
    3.
  3. Torch is zero
    4.
  4. Torch first nonzero
    5.
  5. Torch permute
    6.
  6. Torch take
    7.
  7. Pytorch count nonzero
    8.
  8. Torchargmax
    9.