240

问答题 240 /376

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

参考答案

参考回答:

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