pytorch报错：IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor t

pytorch报错：IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number是你的torch版本的不同造成的。解决：将loss.data[0] 改成loss.item()参考IndexError: invalid index of a 0-d

小枫小疯

799人浏览 · 2021-09-30 18:47:24

小枫小疯 · 2021-09-30 18:47:24 发布

pytorch报错：IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
是你的torch版本的不同造成的。
解决：将loss.data[0] 改成loss.item()

参考 IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python_摩根0的博客-CSDN博客

https://edu.csdn.net/learn/39067/627173?utm_source=2019755004

汇聚全球AI编程工具，助力开发者即刻编程。

更多推荐

Claude Code创始人独家分享：13条神级技巧，彻底改变AI编程思维！

本文深入解析了Claude Code创始人Boris Cherny亲自分享的13条核心使用技巧，揭示了如何将AI编程从简单的对话交互升级为高效的智能协作系统。这些技巧涵盖了并行处理、知识沉淀、工作流自动化、团队协作和智能验证等关键维度，代表了对未来软件开发范式的深刻洞察。通过实践这些方法，开发者的编码效率可获得数倍提升，同时代码质量与团队协作水平也将大幅增强。关键在于转变思维：从"人指挥AI"变为