pytorch 中tensor在CPU和GPU之间转换


1. CPU tensor转GPU tensor:

cpu_imgs.cuda()
2. GPU tensor 转CPU tensor:

gpu_imgs.cpu()
3. numpy转为CPU tensor:

torch.from_numpy( imgs )
4.CPU tensor转为numpy数据:

cpu_imgs.numpy()
5. note:GPU tensor不能直接转为numpy数组,必须先转到CPU tensor。

6. 如果tensor是标量的话,可以直接使用 item() 函数(只能是标量)将值取出来:

print loss_output.item()


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM