textEmbedRawLayers function

Extract layers of hidden states