Nodes' hidden representations?

[chriswtanner]

Hello, not an issue, but what's the easiest way to extract the learned hidden embeddings for each node in a ChildSum tree? New to PyTorch, so forgive my ignorance.

Thanks!