python中cuda()作用,两个Tensor,一个加了cuda(),一个没加,相加后很怎样?

发布于 2020-01-31 16:45:11
关注者
0
被浏览
1787
1 个回答
  • 面试哥
    面试哥 2020-01-31
    为面试而生,有面试问题,就找面试哥。

    参考回答:

    cuda()将操作对象放在GPU内存中,加了cuda()的Tensor放在GPU内存中而没加的Tensor放在CPU内存中,所以这两个Tensor相加会报错。

知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看