ValueError: only one element tensors can be converted to Python scalars

报错# ValueError: only one element tensors can be converted to Python scalars

在dataloader进行保存图片时报错ValueError: only one element tensors can be converted to Python scalars

原文代码如下

报错内容如下

不知道你这个问题是否已经解决, 如果还没有解决的话:

• 建议你看下这篇博客👉 ：only one element tensors can be converted to Python scalars

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^

ValueError：只有一个元素张量可以转换为Python标量，转换错误了
使用 numpy () 方法来将张量转换为 numpy 数组，然后再使用 item () 方法来获取数组中的某个元素