extract embedding

[LAndersen1]

Is there a method to extract an embedding for a protein-ligand complex? E.g. a method that returns the values of the last (or smallest) dense layer