Is there any way to give attention on Embeddingbag?
In other words, the current implementation of Embeddingbag sums up or computes the mean vector of multiple indices given.
What I want to do is, instead of simple mean or sum, I want to compute weighted sum.
So, instead of 1/3*(e1+e2+e3), i want to do the following:
w1*e1+w2*e2+w3*e3
Can anyone help?