如何将tensorflow特定变量作为ndarray保存到本地磁盘?(不是图层参数,只是变量或张量)

时间:2017-04-02 13:55:22

标签: tensorflow deep-learning

如何将tensorflow特定变量作为ndarray保存到本地磁盘?(不是图层参数,只是变量或张量) 喜欢:

ux=tf.Variable([10,1600,1,2])

tf.save('ux.npy',ux)

上面有什么代码吗?之后我可以加载ux.npy,如:

ux = numpy.load('ux.npy')

1 个答案:

答案 0 :(得分:0)

Numpy已经保存并加载了函数,因此为了保存numpy数组,您可以使用它的功能。语法是:

.*