For example if I have a graph of 3 nodes A, B, and C. What I want to do is to feed this graph into a node2vec model and learn an embedding of each node. Here is the pipe line

Construct a graph using nextworkx
Convert it to a pytorch graph data using from_networkx
Construct a node2vec model and train it.
After training I should get a 3XN matrix where N is the embedding dimension I desired. My question is, how do I know which of the 3 embedding is the embedding for node A