@SimonW Do you know of a way to access an embedding multiple times?  I’m building a language model I’d like to target the word vectors of my output directly to save memory.
To do that I’d need both the input vector and the output vector out of the embedding.