Which torch operator can cause nonzero problems?

[980202006]

There is no torch.nonzero in my torch model, but a nonzero error is reported. Which torch operations may cause nonzero problems, or how can I find out which operators contain nonzero operations?

[zhiqwang]

torch.where I guess

[zerollzeng]

I think it's better to ask in the pytorch repo.

[wugoukanle]

it has 3 scene：

1. tensor mask op，as example：x = x[x.sum(dim=-1) > 0]
2. torch.where(condition), as example: torch.where(x>0)
3. torch.nonzero(x)

[980202006]

Thank you!