How to use DGLBlock?

Neighbor sampling in graphbolt results in the list of DGLBlock
I want instead of using embedding from the final layer somehow aggregate embeddings from all layers.
So I need to find the indexes of the final layer output nodes in all blocks outputs. How can I do it? Thank you for assistance.