Thank you for your timely reply.
I apologize for misleading you with the example.
The code is just an example, What I want to do is passing message along batch dimension. Actually, the link is dynamic during processing.
So what should I do if I want to manipulate the feature along batch dim.
In addition, what’s the meaning of leaf Variables?
Best,
Yikang