本文介绍了在numpy中进行数据类型转换时遇到的问题，特别是从numpy数组转换为bytes，然后再转换回来时，维度发生变化的原因。关键在于原始数据类型为float16，而默认转换回numpy数组时使用的数据类型为float64。通过指定正确的dtype可以避免这个问题。解决方案是在使用np.frombuffer时，指定原始数据类型以保持维度一致。
读取一个numpy文件后，把它转成bytes，再转回np，结果发现维度变了！！
例子：
a = np.load('xxx.npy') # (24, 768)
b = a.tobytes()
c = np.frombuffer(b).reshape(-1, 768) # (6, 768)
检查后发现是因为原始numpy的数据类型为float16，而np.frombuffer默认数据类型是np.float64。二者使用相同数据类型即可：
a = np.load('xxx.npy') # (24, 768)
b = a.tobytes()
c = np.frombuffer(b, dtype=a.dtype).reshape(-1, 768) # (24, 768)
扫一扫
818
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
