python - 查找张量中沿轴的非零元素数

Question

我想找出张量中沿特定轴的非零元素的数量。是否有任何 PyTorch 函数可以执行此操作？

我尝试使用 nonzero() PyTorch 中的方法。

torch.nonzero(losses).size(0)

这里，lossess 是形状为 64 x 1 的张量。当我运行上面的语句时，出现以下错误。

TypeError: Type Variable doesn't implement stateless method nonzero

但是如果我运行 torch.nonzero(losses.data).size(0)，那么它就可以正常工作。任何线索，为什么会发生这种情况或错误意味着什么？

Answer 1

错误消息的含义 - TypeError: Type Variable doesn't implement stateless method nonzero 是，我们不能在 autograd 上使用 torch.nonzero()。变量，但仅限于简单的张量。另外需要注意的是，张量是无状态的，而变量是有状态的。