有没有办法在tf.Model对象中处理张量?

时间:2018-04-16 16:27:22

标签: javascript tensorflow.js

我正在构建一个交互式应用程序,用户可以在其中使用模型的参数。指定新参数后,我不再需要旧的tf.Model对象和相关的张量。对于tf.Model,是否有等效的tf.Tensor.dispose()?

谢谢!

3 个答案:

答案 0 :(得分:0)

我能想到的最接近的是tf.reset_default_graph(),虽然它会摧毁所有操作和张量,以及所有变量。

答案 1 :(得分:0)

如果将整个块包裹在tf.tidy()中,它将处理除块返回的所有张量之外的所有张量。

https://js.tensorflow.org/api/0.9.0/#tidy

答案 2 :(得分:0)

也许,tf.disposeVariables()就是你想要的。

此命令在0.11.1版本之后可用。

https://js.tensorflow.org/api/0.11.1/#disposeVariables