你即將離開本站

並前往https://stackoverflow.com/questions/60817261/find-indices-of-elements-equal-to-zero-in-a-pytorch-tensor

Find indices of elements equal to zero in a PyTorch Tensor

You can make a boolean mask and then call nonzero() : (arr == 0).nonzero(). For instance: arr = torch.randint(high=2, size=(3, 3))

確定！ 回上一頁

查詢 「Torch is zero」的人也找了：

  1. Torch is zero
    2.
  2. Torch randperm
    3.
  3. Torch empty
    4.
  4. Torch slice
    5.
  5. Torch nonzero
    6.
  6. Torch flip
    7.
  7. Torch meshgrid
    8.
  8. Torch sum
    9.