如何将 numpy 张量保存到文件中

Question

我正在尝试将图像转换为 4 维的 numpy 张量，我想将其保存到文件中，最好是 csv 文件。我已经有了我想要的尺寸的图像数组，但现在问题是保存到文件中。请问我该怎么做呢。

Answer 1

Numpy 对数组数据的 IO 有很多选择：

• 如果二进制格式没问题，您可以使用np.save 将 4D 张量保存为二进制（“.npy”）格式。可以使用 np.load 再次读取该文件。这是保存 numpy 数据的一种非常方便的方法，它适用于任意维数的数值数组。

• np.savetxt 可以以类似 CSV 的文本格式写入一维或二维数组。您可以使用 np.reshape 将张量展平为 1D 或 2D，然后使用 np.savetxt。缺点是该文件不跟踪完整的 4D 形状，因此您需要以某种方式单独跟踪它。

• 如果以文本表示形式存储很重要，更好的选择可能是使用np.array2string 将张量转换为字符串，然后将字符串写入文件。这甚至适用于超过 2 维的数组。

• .tofile method 只是将元素数据转储为原始二进制文件。不保留任何形状或其他元数据，但二进制文件很容易读入其他程序。