RuntimeError: Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should b

闲看庭前雪

于 2021-06-05 17:07:14 发布

阅读量7.3k

点赞数 4

CC 4.0 BY-SA版权

分类专栏：代码小结神经网络文章标签： python cuda 深度学习神经网络

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/qq_43605708/article/details/117600112

代码小结同时被 2 个专栏收录

15 篇文章

订阅专栏

13 篇文章

订阅专栏

在实现手搓神经网络时遇到RuntimeError，原因是输入数据类型与权重数据类型不一致。问题根源在于输入数据未转换为浮点型。解决方案是在输入数据x前添加.type(torch.cuda.FloatTensor)进行类型转换，并确保使用.cuda()处理GPU设备。参考文章提供了关键思路，实践中发现实际操作还需包括.cuda()来完成数据迁移。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

项目场景：

手搓神经网络

问题描述：

RuntimeError: Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should be the same

原因分析：

没有把输入数据类型转为float类型

解决方案：

x = x.type(torch.cuda.FloatTensor)

注意，加上 .cuda

我参考的文章给了我思路，在我实践的时候，发现还需要添加.cuda

参考链接
链接: link.

评论

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。