Hi @SHAN ,
Your issue seems related to the issue mentioned in this other thread. As mentioned in post #6 in that thread, using qml.transforms.broadcast_expand might help you.
That being said, I don’t think qml.AmplitudeEmbedding was designed to be used more than once so the solution I mentioned may not work in this case. Why don’t you give it a try and let us know if this works for you?
I hope this helps!