Missing padding_idx option in EmbeddingBag

1

I have to deal with variable length of indices of embeddings, in which EmbeddingBag is perfect for me.
Now, I need to ignore some embeddings in one bag with probabilities. I tried to use padding_idx then I found there is no such option in EmbeddingBag. (I can map these discarded index to padding_index.)
Is this because of some implementation restrict ?
How can I ignore some embedding in one bag while keeps using EmbeddingBag?
Any help will be appreciated.
Thanks

1 Like
2

Any progress on this issue?
Actually, Iā€™m having a similar problem so any comments are greatly appreciated.

1 Like