不幸的是,tf.group()不返回所计算操作的输出。我正在寻找一种方法来执行tf.group()的工作,但要获取那些张量的输出。 我希望可以在图表中完成所有操作,这样我就不必使用feed dict。
发生这种情况的情况是,如果您要创建一个包含每个批处理的行号的文本解析器数据集。
start = 10
ts = tf.Variable(start,dtype=tf.int32)
ts_p1 = ts.assign_add(1)
ds = tf.data.TextLineDataset(filenames[0])
ds = ds.skip(start+1)
# Below is the line of interest
ds = ds.map(lambda line: tf.group([ts_p1,line]))
ds = ds.batch(1)
ds = ds.repeat()
it = ds.make_initializable_iterator()
nxt = it.get_next()