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

我想找到沿特定轴的张量中非零元素的数量.有没有可以执行此操作的PyTorch函数？

我试图在PyTorch中使用nonzero()方法.

torch.nonzero(losses).size(0)

在这里,损失是一个形状为64 x 1的张量.当我运行上面的语句时,它给了我以下错误.

TypeError: Type Variable doesn't implement stateless method nonzero

但是如果我运行torch.nonzero(losses.data).size(0),那么它可以正常工作.有任何线索,为什么会这样或错误意味着什么？

解决方法:

错误消息的含义-TypeError：类型变量没有实现非零状态的非零方法,我们不能在autograd.Variable上使用torch.nonzero(),而只能在简单张量上使用.还应注意,张量是无状态的,而变量是有状态的.

标签：pytorch,deep-learning,python
来源： https://codeday.me/bug/20191111/2018284.html