首页
题库
面试
求职
学习
竞赛
More+
所有博客
搜索面经/职位/试题/公司
搜索
我要招人
去企业版
登录 / 注册
首页
>
算法/机器学习校招面试题目合集
>
pytorch中cuda()作用,两个Tensor,一个加了cuda(),...
240
问答题
240
/
376
pytorch中cuda()作用,两个Tensor,一个加了cuda(),一个没加,相加后很怎样?
参考答案
参考回答:
cuda()将操作对象放在GPU内存中,加了cuda()的Tensor放在GPU内存中而没加的Tensor放在CPU内存中,所以这两个Tensor相加会报错。
纠错
收藏
查看讨论
1
...
235
236
237
238
239
240
241
242
243
244
245
...
376
跳转到
确 定
上一题
下一题
扫描二维码,关注牛客网
意见反馈
下载牛客APP,随时随地刷题