learned_embeddings = model.embeddings.word_embeddings(token_ids)
learned_embeddings