解决UserWarning: Creating a tensor from a list of numpy.ndarrays is extremely slow.

📚博客主页:knighthood2001

✨公众号:认知up吧 (目前正在带领大家一起提升认知,感兴趣可以来围观一下)

🎃知识星球:【认知up吧|成长|副业】介绍

❤️感谢大家点赞👍🏻收藏⭐评论✍🏻,您的三连就是我持续更新的动力❤️

🙏笔者水平有限,欢迎各位大佬指点,相互学习进步!


报错如下:
UserWarning: Creating a tensor from a list of numpy.ndarrays is extremely slow. Please consider converting the list to a single numpy.ndarray with numpy.array() before converting to a tensor.

这个警告是由于在使用 torch.FloatTensor() 创建张量时,传入了一个包含多个 numpy 数组的 Python 列表,而这种方式相对较慢。
警告建议你在转换为张量之前,将列表转换为一个单一的 numpy 数组,然后再进行张量转换。

我代码中的这一行触发了警告:

b_s = torch.FloatTensor(b_s)  # 取出s

在这里,b_s 是一个包含多个 numpy 数组的列表。为了消除警告,你可以使用 numpy.array() 将列表转换为单一的 numpy 数组,然后再将其转换为 PyTorch 张量。修改后的代码如下:

b_s = torch.FloatTensor(np.array(b_s))  # 取出s

这样做可以提高代码执行效率并消除警告。在实际应用中,如果你的 b_s 列表元素的数据类型一致,将其转换为一个 numpy 数组通常是更高效的做法。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

knighthood2001

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值