Pytorch 可微计数
我需要计算满足条件的张量中元素的数量,例如计算年龄==60岁或年龄>=50岁的人数。计数函数有可微近似吗?使用Pytorch 可微计数,pytorch,mathematical-optimization,Pytorch,Mathematical Optimization,我需要计算满足条件的张量中元素的数量,例如计算年龄==60岁或年龄>=50岁的人数。计数函数有可微近似吗?使用火炬张量.sum或使用年龄>=50返回的布尔张量 age = torch.arange(75) (age >= 50).sum() tensor(25)
火炬张量.sum
或使用年龄>=50返回的布尔张量
age = torch.arange(75)
(age >= 50).sum()
tensor(25)