在aixs中找出张量中非零元素的数量

Question

我想找到沿特定轴的张量中非零元素的数量。有没有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)，那么它运作正常。任何线索，为什么会发生这种情况或错误意味着什么？

Answer 1

错误消息的含义 - TypeError: Type Variable doesn't implement stateless method nonzero是，我们不能在torch.nonzero()上使用autograd.Variable，而只能在简单的张量上使用pandoc。还应该注意的是，张量是无状态的，而变量是有状态的。