Pytorch中 detach() item() cpu() numpy()互相转换 2. pytorch 中tensor在CPU和GPU之间转换,以及numpy之间的转换:

参考链接:

1. Pytorch中 detach() item() cpu() numpy()理解:

https://blog.****.net/weixin_43289424/article/details/105803097

https://blog.****.net/moshiyaofei/article/details/90519430

3. Pytorch cuda上的tensor转numpy:

https://blog.****.net/lxb206/article/details/113773886

GPU tensor不能直接转为numpy数组,必须先转到CPU tensor。