【解决办法】报错Found dtype Long but expected Float

Found dtype Long but expected Float错误通常发生在尝试将一个数据类型为Long的张量传递给一个期望数据类型为**Float**的函数或操作时。

在PyTorch中,Long 和**Float**是两种常见的数据类型,分别对应于64位整数和32位浮点数。某些函数或操作可能只接受特定数据类型的输入。例如loss_func函数出现报错,它需要输入数据为float

这时候只需要通过.float将数据类型转为float即可满足loss_func的输入数据类型要求

复制代码
        out_dis = out_dis.float()
        yb = yb.float()
        loss_r = loss_func(out_dis, yb)
        loss_r.backward()

需要注意的是这种转换可能会导致数据丢失,进而影响后续的计算,如这里的损失函数,因为Long类型的整数可以表示的范围比Float类型的浮点数大。

相关推荐
pen-ai2 小时前
【高级机器学习】 12. 强化学习,Q-learning, DQN
人工智能·机器学习
受之以蒙2 小时前
Rust ndarray 高性能计算:从元素操作到矩阵运算的优化实践
人工智能·笔记·rust
fanjinhong_85212 小时前
屏幕捕捉工具 (Screen Capture Tool)
python·github
野生面壁者章北海2 小时前
NeurIPS 2024|大语言模型高保真文本水印新范式
人工智能·语言模型·自然语言处理
KG_LLM图谱增强大模型2 小时前
如何利用大语言模型(LLM)实现自动标注与内容增强
人工智能·知识管理·内容管理·本体论·图谱增强大模型·自动标签·大模型内容标注
数据与后端架构提升之路3 小时前
小鹏VLA 2.0的“神秘涌现”:从痛苦到突破,自动驾驶与机器人如何突然“开窍”?
人工智能·机器人·自动驾驶
fruge3 小时前
CANN核心特性深度解析:简化AI开发的技术优势
人工智能
沛沛老爹3 小时前
AI入门知识之RAFT方法:基于微调的RAG优化技术详解
人工智能·llm·sft·raft·rag
zskj_zhyl3 小时前
科技助老与智慧养老的国家级政策与地方实践探索
大数据·人工智能·科技