首页 > 试题广场 >

pytorch中cuda()作用,两个Tensor,一个加了

[问答题]

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

添加回答
这次一定要上岸头像 这次一定要上岸
会报错 类型不匹配。默认数据是传递到cpu上的，加上cuda()会将数据传递到gpu上，两者相加会报错。
发表于 2020-09-18 19:14:12 回复(0)
提交观点

问题信息

上传者：小小
难度：
1条回答 1245浏览

热门推荐

相关试题

扫一扫，把题目装进口袋

求职之前，先上牛客

扫描二维码，进入QQ群

扫描二维码，关注牛客公众号