参考链接：
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。