将张量值保存为二进制格式的最佳方法是什么？

Question

我试图将张量值保存为二进制格式。 特别是我试图将float32张量值保存为二进制格式（IEEE-754格式）。你能帮帮我吗？？

import tensorflow as tf

x = tf.constant([[1.0, 2.0, 3.0], [5.5, 4.3, 2.5]])

# how to save tensor x as binary format ?? 

Answer 1

建议的方法是检查模型。如Saving and Restoring programmer's guide中所述，创建^{}对象，可以选择指定要保存的变量/可保存对象。然后，每当您想保存张量的值时，就调用tf.train.Saver对象的save（）方法：

saver = tf.train.Saver(...)

#...

saver.save(session, 'my-checkpoints', global_step = step)

。。其中第二个参数（'my-checkpoints'在上面的示例中）是存储检查点二进制文件的目录的路径。

另一种方法是计算单个张量（将是NumPy ndarrays），然后将单个ndarrays保存到NPY文件（通过^{}）或多个ndarrays保存到单个NPZ存档（通过^{}或^{}）：

np.save('x.npy', session.run(x), allow_pickle = False)