Pytorch: Differentiable counting

Question

I need to count the number of elements in a tensor that satisfy a condition, such as counting the number of people with age == 60, or people with age >= 50. Is there a differentiable approximation to the counting function?

Dishin Goyani · Accepted Answer · 2020-07-08 06:12:10Z

1

Use torch.Tensor.sum or torch.sum on boolean tensor returned by age >= 50

age = torch.arange(75)

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

answered Jul 8, 2020 at 6:12

Dishin Goyani

7,7533 gold badges33 silver badges42 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Pytorch: Differentiable counting

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related