My input is a list of 5 embeddings and I want to predict whether the 5 entities are selected. If I only need to output one entity, I can use softmax activation. Then what if two of them are selected?
Basically I want a generalized softmax and the sum of the output vector is 2 instead of 1. Also each element of the output is less than or equal to 1.