c - 如何使用低级 (C) 接口(interface)将张量保存到 TensorFlow 中的文件？

Question

使用 TensorFlow 中的低级操作，我尝试将张量(其实际值)保存到磁盘，但找不到如何保存。如果数据是例如一个表示图像的 UInt8 矩阵，然后我可以轻松地执行 EncodeJpeg 来创建内容，然后使用 WriteFile 将生成的内容写入具有给定名称的文件。同样，EncodeWav 的工作方式相同。另一方面，如果我只想保存一个带有数字的矩阵，没有“EncodeData”、“TensorToContent”或类似的操作将 Tensor 转换为 Content，WriteFile 可以保存什么。我可以从我的图形中获取张量作为输出，然后将其保存在图形之外，但我的目的是在图形内部进行。

Answer 1

我花了一些时间，但终于找到了解决方案，尽管它可能不是最好的。如果您能改进，不客气。

因此，我使用保存/恢复原始操作对。一旦找到，它就相对简单。

“Save”有两个张量Inputs，一个张量InputList和一个Attribute。第一个张量(标量字符串)必须给出文件名。第二个张量(一维字符串)包含您要为文件中的张量指定的名称(用于稍后的恢复目的)。 InputList 实际上是要保存的张量的集合。 Attribute 是要保存的张量的数据类型列表。

“恢复”也很简单。它有两个张量输入，一个张量输出，一个类型属性和一个可选的整数属性。如果可以查看更多文件，第一个输入(标量字符串)给出文件名或文件模式。第二个输入(标量字符串)给出了您要在文件中查找的张量的名称(见上文“保存”)。输出是加载的张量本身。 type 属性指定将加载的张量的类型。如果具有相同名称的张量存在于与模式匹配的多个文件中，则可以使用可选属性，告诉哪个是首选。

唯一需要注意的是“Save”中的InputList。当在设计时将“保存”原始操作添加到图形时，必须给出输入列表中的输入数量，以及类型列表参数中的类型(相同数量!)。这意味着在运行时，当给出张量的实际名称并将张量列表作为 InputList 时，您不能再更改张量的数量和数据类型。