📚博客主页: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 数组通常是更高效的做法。