如何将文件中的值分配给TensorFlow变量而不在内存中存储2个副本?

时间:2017-03-02 23:55:13

标签: python tensorflow

know可以使用语句x.assign(1),然后调用Operation.run()Session.run(),将值1分配给TensorFlow变量{{ 1}}。

但是,此解决方案将x两次存储在内存中。当要分配的值很大(例如,单词嵌入)时,这可能是一个问题。

如何将文件中的值分配给TensorFlow变量而不将2个副本值存储在内存中?

1 个答案:

答案 0 :(得分:1)

您可以使用TensorFlow阅读器避免将值存储在Python内存中。 IE,类似

filename_queue = tf.train.string_input_producer("myfile")
x.assign(reader.read(filename_queue))

你想要分配一些东西而不是字符串,你可能需要将reader.read与产生所需类型的op结合起来,即TextLineReader + tf.decode_csv

https://www.tensorflow.org/programmers_guide/reading_data