I’m still unsure, how this method should work and would need to see at least pseudo code for it.
E.g. I don’t know, what [X], [XY, YX] stand for and why the indices change for the latter case suddenly.
I would be helpful, if you could all at least pseudo code with tensors and their shape for each step and what the desired output should look like.