摘要: 报错原因:numpy不能读取CUDA tensor 需要将它转化为 CPU tensor。 所以如果想把CUDA tensor格式的数据改成numpy时,需要先将其转换成cpu float-tensor随后再转到numpy格式 报错行: 修改后: 阅读全文
posted @ 2019-04-09 15:46 小呆丶 阅读(10941) 评论(0) 推荐(1) 编辑