Solved by albanD in post #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.