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?