Pytorch：可微分计数

【问题标题】：Pytorch: Differentiable countingPytorch：可微分计数
【发布时间】：2020-07-08 14:12:10
【问题描述】：

我需要计算张量中满足某个条件的元素的数量，例如计算年龄 == 60 或年龄 >= 50 的人数。计数函数是否存在可微近似？

【问题讨论】：

【解决方案1】：

在age >= 50 返回的布尔张量上使用torch.Tensor.sum 或torch.sum

age = torch.arange(75)

(age >= 50).sum()
tensor(25)

【讨论】：