与非零相反的pytorch是什么

【问题标题】：What is pytorch opposite of nonzero与非零相反的pytorch是什么
【发布时间】：2022-08-18 17:52:57
【问题描述】：

PyTorch 中的 nonzero 返回非零元素的索引。

tensor([ True, False,  True,  True,  True,  True,  True, False,  True, False])
tensor([0, 2, 3, 4, 5, 6, 8])

它的反面是什么？要获得全零的索引？

【解决方案1】：

您可以简单地使用torch.where：

import torch

a = torch.randint(2, size=(8,))
torch.where(a == 0)[0]

【讨论】：

【解决方案2】：

(x == 0).nonzero()

只需改造nonzero 函数。

【讨论】：