• 參考
• 整體流程如下圖
  
• 代碼

import tensorflow as tf"""模擬: 子線程不停的取數(shù)據(jù)放入隊(duì)列中, 主線程從隊(duì)列中取數(shù)據(jù)執(zhí)行包含: 作用域的命名、把程序的圖結(jié)構(gòu)寫入事件、多線程 """# 模擬異步存入樣本. # 1、 定義一個(gè)隊(duì)列,長(zhǎng)度為1000 with tf.variable_scope("init_sub_threads"):Q = tf.FIFOQueue(1000, tf.float32)# 2、 定義子線程需要做的事情, 值循環(huán)+1, 放入隊(duì)列中 with tf.variable_scope("add_one"):var = tf.Variable(0.0, tf.float32)# 實(shí)現(xiàn)自增data = tf.assign_add(var, tf.constant(1.0))en_q = Q.enqueue(data)# 3、 定義隊(duì)列管理器Op,指定子線程 with tf.variable_scope("init_queue_runner"):qr = tf.train.QueueRunner(Q, enqueue_ops=[en_q] * 2)# 初始化變量的op with tf.variable_scope("init_var"):init_op = tf.global_variables_initializer()with tf.Session() as sess:# 初始化變量sess.run(init_op)# 開啟線程管理器coord = tf.train.Coordinator()# 真正開啟子線程threads = qr.create_threads(sess, start = True)# 主線程,不斷讀取數(shù)據(jù)訓(xùn)練for i in range(300):print(sess.run(Q.dequeue()))# 把程序的圖結(jié)構(gòu)寫入事件filewriter = tf.summary.FileWriter("./summary/", graph=sess.graph)# 回收線程coord.request_stop()coord.join(threads)

總結(jié)

以上是生活随笔為你收集整理的[异步、tensorflow] - 子线程操作tensor,主线程处理tensor的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。