什么是并行构建/执行for循环Tensorflow图的好方法?

时间:2018-06-24 18:50:22

标签: tensorflow pytorch

y = []
for i in range(9999):
  x = ... # build/get sub graph
  y.append(x)
y = tf.stack(x)

我正在尝试构建一个Tensorflow图,其中包含多个单独的子图/输入(彼此独立),我正在使用python for循环来实现此目的。 (假设for循环无法通过张量操作实现)。 我发现这部分真的很慢,我认为这是因为python for循环不是并行的(一个接一个地处理它们)。我想知道是否有解决方案可以使它并行(因为每个子图/输入彼此独立且独立)?我不知道更改为tf.while_loop可能对您有帮助还是有其他建议?

0 个答案:

没有答案