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。

   

posted @ 2021-03-30 19:39  vv_869  阅读(1418)  评论(0编辑  收藏  举报