ptrblck:

to(device='cpu') operation will be a no-op on the CPU and a valid operation on the GPU.
You can check it additionally via print(a.is_leaf) and print(b.is_leaf), which will show True on the CPU and False on the GPU.


Awesome! Thank you for the insightful explanation Peter!