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