Hi,
This code looks good to me.
The gradient will flow back from node_embessing to seq_embedding and all the way through your model. Do you have specific observations that make you think that this is an autograd problem?