Python Tensorflow,tf.train.batch,无结果
我对tf.train.batch是个新手,所以我写了一个样本来测试它。当我运行代码时,没有得到任何结果,进程仍在运行 你以前遇到过同样的情况吗?非常感谢Python Tensorflow,tf.train.batch,无结果,python,tensorflow,tensor,Python,Tensorflow,Tensor,我对tf.train.batch是个新手,所以我写了一个样本来测试它。当我运行代码时,没有得到任何结果,进程仍在运行 你以前遇到过同样的情况吗?非常感谢 import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' import numpy as np import tensorflow as tf a = [[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4]] b = [1,2,3,4] input_queue = tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
import numpy as np
import tensorflow as tf
a = [[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4]]
b = [1,2,3,4]
input_queue = tf.train.slice_input_producer([a, b],num_epochs=None,shuffle=False)
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(sess=sess, coord=coord)
for i in range(4):
x,y = tf.train.batch([a,b], batch_size=2)
x_,y_ =sess.run([x,y])
print(x_,y_)
coord.request_stop()
coord.join(threads)
另外,函数tf.train.slice\u input\u producer也可以工作。当我忽略tf.train.batch时,代码变为:
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
import numpy as np
import tensorflow as tf
a = [[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4]]
b = [1,2,3,4]
input_queue = tf.train.slice_input_producer([a, b],num_epochs=None,shuffle=False)
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(sess=sess, coord=coord)
for i in range(4):
print(sess.run(input_queue))
coord.request_stop()
coord.join(threads)
结果是:
[array([1, 2, 3, 4]), 1]
[array([1, 2, 3, 4]), 2]
[array([1, 2, 3, 4]), 3]
[array([1, 2, 3, 4]), 4]
我认为主要的问题是您没有指定将许多排队到
True
,以便it只会重复整个批次。你可以在网站上阅读更多。
以下是一个工作示例:
a = [[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4]]
b = [1,2,3,4]
a, b = tf.train.batch([a,b], batch_size=1, num_threads=1, capacity=4, enqueue_many=True)
with tf.Session() as sess:
coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(sess, coord)
for i in range(4):
print(sess.run([a,b]))
coord.request_stop()
coord.join(threads)
你想要的结果是什么?
tf.train.batch
主要用于输入管道。xiexie dalao l.@Jingjue.Wei不知道你的意思,但如果你认为这个答案解决了你的问题,请点击勾号接受。它似乎有效,但我已经运行了这个代码。如果我有什么问题,我会联系你