我不太明白while_loop并行化是如何工作的。假设我有一个TensorArray,它有10个形状相同的张量。现在假设循环体中对前5个张量的计算独立于其余5个张量的计算。TensorFlow会并行运行这两个任务吗?另外,如果我使用张量而不是TensorArray,并使用scatter_update对其进行更新,它会在backprop过程中正确传递梯度吗?
发布于 2016-12-23 12:55:06
你可能会得到前5次迭代和后5次迭代的并行执行。我可以肯定地说,如果你提供一个代码样本。
https://stackoverflow.com/questions/41273756
复制相似问题