本文介绍了如何将CUDA Tensor数据转换为numpy格式，关键在于通过cpu().numpy()方法实现，避免直接使用numpy报错。
问题分析
如果想把CUDA tensor格式的数据改成numpy时，需要先将其转换成cpu float-tensor随后再转到numpy格式。 numpy不能读取CUDA tensor 需要将它转化为 CPU tensor
解决方法
将报错代码self.numpy()改为self.cpu().numpy()即可
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
