How to get last layer embedding

[rajkumar-surana]

I was trying to get last layer embedding that too of only one token[should i go with first token or I can pool]. How can I do that? I want to get embeddings for query-query or query-passage similarity tasks.. Thanks