Yeah, I mean this line of code.
I can’t currently test it and I know it was an in-place operation in the old versions.
Now I’m wondering, if it’s just a call to tensor.to, which would make the assignment necessary.