我know可以使用语句x.assign(1)
,然后调用Operation.run()
或Session.run()
,将值1
分配给TensorFlow变量{{ 1}}。
但是,此解决方案将x
两次存储在内存中。当要分配的值很大(例如,单词嵌入)时,这可能是一个问题。
如何将文件中的值分配给TensorFlow变量而不将2个副本值存储在内存中?
答案 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