x is a large matrix of size with 1 million rows and 1000 columns.
Doing math operation is very fast as expected. But retrieve the result, which is a Tensor of size 1 takes more than 10 sec. What is happening here.

pytorch_time_move_to_cpu.png750×485 38.8 KB