Retrieve feature embeddings

[johanenforcer]

Im not quite sure how I should go about retrieving feature embeddings (activation scores) during inference from lets say the penultimate layer prior to the final output layer, any ideas?

Thanks in advance