How to get word embedding from a TF bert model?

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