在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 ，我們不能在autograd.Variable上使用torch.nonzero() ，但只能在簡單的張量上使用。 還應該注意的是，張量是無狀態的，而變量是有狀態的。