djseivad:

So the reassignment of inputs acts like a pointer to a device memory space containing the tensors?


Tensors return a copy on the specified device, while model.to() works recursively on all submodules.
How large is your embedding layer?