Pytorch如何在张量中查找非零元素

我想要沿特定的轴找到张量的非零元素的数量。有没有可以做到这一点的PyTorch功能?
我试图在PyTorch中使用nonzero()方法。
torch.nonzero(losses).size(0)
这里lossess是一个张量的形状64 x 1。当我运行上面的语句,它给我以下错误。
TypeError: Type Variable doesn't implement stateless method nonzero
但是,如果我跑,torch.nonzero(losses.data).size(0)那么它工作正常。

任何线索,为什么会发生这种情况?
已邀请:

要回复问题请先登录注册