沿轴查找张量中非零元素的数量答案

【问题标题】：Find number of non-zero elements in a tensor along an aixs沿轴查找张量中非零元素的数量
【发布时间】：2017-12-05 01:23:30
【问题描述】：

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

【问题讨论】：

标签： python deep-learning pytorch

【解决方案1】：

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

【讨论】：