Pytorch中 detach() item() cpu() numpy()互相转换
参考链接:
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。