Will moving tensor from GPU to CPU to GPU break back-propagation?

Hi,

No moving Tensors across devices is perfectly differentiable. It will all work fine!

3 Likes