參考鏈接：
1. Pytorch中 detach() item() cpu() numpy()理解：
https://blog.csdn.net/weixin_43289424/article/details/105803097
2. pytorch 中tensor在CPU和GPU之間轉換，以及numpy之間的轉換：
https://blog.csdn.net/moshiyaofei/article/details/90519430
3. Pytorch cuda上的tensor轉numpy：
https://blog.csdn.net/lxb206/article/details/113773886
GPU tensor不能直接轉為numpy數組，必須先轉到CPU tensor。