I face this problem too and I followed you solution too, but in my first attempt as a workaround I tried to clone the tensor into a new variable but it didn’t work.
Do you plan to support this in the near future?
Also, I think that you have a typo (^2 should be **2)