我想找到沿特定轴的张量中非零元素的数量。有没有 PyTorch 函数可以做到这一点？
我尝试使用非零() http://pytorch.org/docs/master/torch.html?highlight=nonzero#torch.nonzeroPyTorch 中的方法。
torch.nonzero(losses).size(0)
Here, lossess是一个形状张量64 x 1。当我运行上面的语句时，它给出了以下错误。
TypeError: Type Variable doesn't implement stateless method nonzero
但如果我跑的话torch.nonzero(losses.data).size(0)，然后就可以正常工作了。任何线索，为什么会发生这种情况或者错误意味着什么？
错误消息的含义 -TypeError: Type Variable doesn't implement stateless method nonzero是，我们不能使用torch.nonzero() on autograd.Variable但仅限于简单的张量。还应该注意的是，张量是无状态的，而变量是有状态的。
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
