Does .item() automatically move the data to the cpu?

2

Hi,

Yes .item() moves the data to CPU. It converts the value into a plain python number. And plain python number can only live on the CPU.

7 Likes