Hi, I’ve been investigating this issue these days. The approach you provided seems working properly. Thank you so much!!!
However, I also tried something like this:
y = wx
y1 = w1x1
y.data = y1.data

From my understanding, this should also work. But it actually led to different result from your approach. Do you know why is that?