I want to get first layer word embedding output, not word_embeddings weights. can someone help me? thx.