What is nn.embedding exactly doing?

It is just a look up table from indices to vectors. You can manually initialize them however you want, e.g. to work2vec weights.

5 Likes