To give an example of what I need, say I have the following labels as target:
tensor([2, 8, 2, 6])

I would like to replace each label with the corresponding row of the embeddings I have. I think this means ending up with a 2D tensor like:
tensor(
[[0.0902, 0.0642, 0.1733, 0.1177, 0.0837, 0.0939, 0.0653, 0.1129, 0.0894, 0.1095],
 [0.1019, 0.1095, 0.0896, 0.0903, 0.0637, 0.0944, 0.0465, 0.1101, 0.1737, 0.1204],
 [0.0902, 0.0642, 0.1733, 0.1177, 0.0837, 0.0939, 0.0653, 0.1129, 0.0894, 0.1095],
 [0.1262, 0.1171, 0.0811, 0.0780, 0.1252, 0.0888, 0.2153, 0.0586, 0.0576, 0.0521]]
)