import torch
import torch.nn as nn
emb = nn.Embedding(num_embeddings=10, embedding_dim=3) # M de taille 10 x 3
ids = torch.tensor([[1, 5, 5, 2]]) # 1 séquence de 4 tokens, shape (1, 4)
x = emb(ids)
print(x.shape) # torch.Size([1, 4, 3])
print(torch.equal(x[0, 0], emb.weight[1])) # True : le token 1 donne la ligne 1 de M