TensorFlow张量到TensorFlow.js张量

Question

我试图从ConvNet输出中获取张量，并在TF.js中的另一个模型中使用它。有什么方法可以将TF张量保存到文件并在TF.js中加载该文件并恢复张量吗？

Answer 1

张量可以保存到文件中。但是，要将其加载回服务器是必需的，因为除非使用input type file，否则浏览器不会直接访问文件系统。

要将张量保存到文件中，必须先使用data或dataSync下载张量值。

const tensor = tf.tensor([1, 2])
// download values
const values = tensor.dataSync()

要查看如何将values保存到文件，可以使用以下answer

要加载保存的张量，文件内容需要由带有http请求的服务器提供。收到请求的响应后，

const tensorRetrieved = tf.tensor(contentHttpResponse) // if needed parse the string response using JSON.parse

另一个选择是在节点上使用tensorflow.js。由于节点可以访问文件系统，因此不需要http请求和响应。