TensorFlow while_loop并行化TensorArray

Question

我并不完全了解while_loop并行化的工作原理。假设我有一个TensorArray，它有10个具有相同形状的张量。现在假设前5个张量的循环体中的计算独立于其余5个张量中的计算。 TensorFlow会并行运行这两个吗？此外，如果我使用Tensor而不是TensorArray并使用scatter_update对其进行更新，它会在backprop期间正确传递渐变吗？

Answer 1

您应该可以获得前5次迭代和后5次迭代的并行执行。我可以肯定地说你是否提供了代码示例。