Python 沿aix查找张量中非零元素的数量

我想找到张量中沿着特定轴的非零元素的数量。有任何Pytork功能可以做到这一点吗

我试着在PyTorch中使用这种方法

torch.nonzero(losses).size(0)

这里，

lossess

是一个形状为

64x1

的张量。当我运行上面的语句时，它给出了以下错误

TypeError: Type Variable doesn't implement stateless method nonzero

---

但是如果我运行，

torch.nonzero（loss.data）.size（0）

，那么它就可以正常工作。任何线索，为什么会发生这种情况，或者错误的含义是什么？

错误消息的含义-

TypeError:Type变量没有实现无状态方法nonzero

是，我们不能在

autograd.Variable

上使用

torch.nonzero（）

，而只能在简单张量上使用。还应该注意的是，张量是无状态的，而变量是有状态的